提交 54d767c1 authored 作者: Frederic's avatar Frederic 提交者: Marc-Alexandre Cote

Disable the perform on the gpu op.

上级 db7c8b58
......@@ -39,6 +39,9 @@ class GpuCumsum(CumsumOp, GpuOp):
self.max_threads_dim0 = None
self.max_grid_size1 = None
def perform(self, node, inp, out):
return Op.perform(self, node, inp, out)
def make_node(self, x):
assert x.dtype == 'float32'
if not isinstance(x.type, CudaNdarrayType):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论