提交 fa37f260 authored 作者: Frederic's avatar Frederic

Fix crash in debugmode

上级 fa8c474c
......@@ -3070,7 +3070,7 @@ class GpuJoin(tensor.Join, GpuOp):
axis, cndas = axis_and_tensors[0], axis_and_tensors[1:]
# In case axis is numpy.int8 and has no __index__() method
axis = int(axis)
ndim = tensors[0].ndim
ndim = cndas[0].ndim
if axis < -ndim:
raise IndexError("Join axis %d out of bounds [0, %d)" %
(axis, ndim))
......
......@@ -245,7 +245,7 @@ class Scan(PureOp):
"that it shouldn't be the case")
for out in self.outputs:
if isinstance(inp.type, GpuArrayType):
if isinstance(out.type, GpuArrayType):
raise TypeError("Inconsistency in the inner graph of "
"scan '%s' : one of the outputs to the "
"inner graph is of type GpuArrayType but "
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论