提交 9256890c authored 作者: Frederic's avatar Frederic

small code simplication.

上级 a1bad4b4
......@@ -3289,7 +3289,7 @@ class GpuContiguous(GpuOp):
Py_INCREF(%(z)s);
} else if ((NULL == %(z)s)""" % locals()
for i in xrange(len(node.inputs[0].type.broadcastable)):
for i in xrange(node.inputs[0].type.ndim):
str += "\n|| (CudaNdarray_HOST_DIMS(%(input)s)[%(i)s] != CudaNdarray_HOST_DIMS(%(z)s)[%(i)s])" % locals()
str += """
|| !CudaNdarray_is_c_contiguous(%(z)s))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论