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

Remove DebugMode warning

上级 b7f4733f
...@@ -25,8 +25,9 @@ class GpuCumsum(CumsumOp, GpuOp): ...@@ -25,8 +25,9 @@ class GpuCumsum(CumsumOp, GpuOp):
self.max_grid_size1 = None self.max_grid_size1 = None
self.max_grid_size2 = None self.max_grid_size2 = None
def perform(self, node, inp, out): # We must reuse the same method, not reimplement and call it.
return Op.perform(self, node, inp, out) # Otherwise DebugMode will print many warnings.
perform = Op.perform
def make_node(self, x): def make_node(self, x):
assert x.dtype == 'float32' assert x.dtype == 'float32'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论