提交 0d786d08 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix problems.

上级 0d867ade
......@@ -1620,7 +1620,7 @@ class GpuDnnReduction(DnnBase):
bcast = []
for i in range(inp.ndim):
if not (self.axis & (1 << i)):
if not (self.c_axis & (1 << i)):
bcast.append(inp.broadcastable[i])
outs = [inp.type.clone(dtype=self.dtype, broadcastable=bcast)()]
if self.arg:
......
......@@ -17,6 +17,7 @@ from ..basic_ops import (
from ..blas import GpuGemm
from ..elemwise import (
GpuCAReduceCuda, GpuCAReduceCPY, GpuElemwise, Elemwise, max_inputs_to_GpuElemwise)
from ..dnn import GpuDnnReduction
from ..subtensor import GpuSubtensor
from ..linalg import GpuCusolverSolve, cusolver_available, GpuCholesky
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论