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

Fix crash in debugmode

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