提交 8b422739 authored 作者: Frederic's avatar Frederic

Fix error reporting.

上级 3084a1f2
......@@ -2987,6 +2987,11 @@ class GpuJoin(tensor.Join, GpuOp):
shape_%(cdna)s[i] = CudaNdarray_HOST_DIMS(%(cdna)s)[i];
if((i!=axis) && (shape_%(cdna)s[i]!=shape_out[i]))
{
PyErr_Format(
PyExc_ValueError,
"GpuJoin: Wrong inputs for input %%d related"
" to inputs 0.!",
i);
%(fail)s;
}
}
......
......@@ -4307,7 +4307,7 @@ class T_Join_and_Split(unittest.TestCase):
def test_broadcastable_flags_many_dims_and_inputs(self):
"""
Test that the right broadcastable flags get set for a join
Test that the right broadcastable flags get set for a join
with many inputs and many input dimensions.
"""
a = TensorType(dtype=self.floatX, broadcastable=[1, 0, 1, 0, 0, 0])()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论