提交 19b2551e authored 作者: Frederic's avatar Frederic

Fix pycuda test in FAST_COMPILE

上级 a1eef12d
......@@ -3345,6 +3345,13 @@ class GpuContiguous(GpuOp):
input = as_cuda_ndarray_variable(input)
return Apply(self, [input], [input.type()])
def perform(self, node, inp, out):
i = inp[0]
if not i.is_c_contiguous():
i = i.copy()
assert i.is_c_contiguous()
out[0][0] = i
def c_code(self, node, name, inp, out, sub):
input, = inp
z, = out
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论