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

Add c code for HostFromGpu.

上级 1bf79d05
...@@ -79,6 +79,19 @@ class HostFromGpu(GpuOp): ...@@ -79,6 +79,19 @@ class HostFromGpu(GpuOp):
def infer_shape(self, node, xshp): def infer_shape(self, node, xshp):
return xshp return xshp
def c_code(self, node, name, inputs, outputs, sub):
inp = inputs[0]
out = outputs[0]
fail = sub['fail']
return """
%(out)s = (PyArrayObject *) CudaNdarray_CreateArrayObj(%(inp)s);
if(!%(out)s)
%(fail)s;
""" % locals()
def c_code_cache_version(self):
return (1,)
host_from_gpu = HostFromGpu() host_from_gpu = HostFromGpu()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论