提交 fe4df8b5 authored 作者: notoraptor's avatar notoraptor

Avoid usage of CPU C code in sandbox.cuda.basic_ops.GpuSplit.

上级 50911749
...@@ -3599,7 +3599,7 @@ class GpuJoin(tensor.Join, GpuOp): ...@@ -3599,7 +3599,7 @@ class GpuJoin(tensor.Join, GpuOp):
gpu_join = GpuJoin() gpu_join = GpuJoin()
class GpuSplit(tensor.Split, GpuOp): class GpuSplit(gof.HideC, tensor.Split, GpuOp):
def make_node(self, x, axis, splits): def make_node(self, x, axis, splits):
x = as_cuda_ndarray_variable(x) x = as_cuda_ndarray_variable(x)
node = tensor.Split.make_node(self, x, axis, splits) node = tensor.Split.make_node(self, x, axis, splits)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论