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