提交 ff9c346e authored 作者: abergeron's avatar abergeron

Merge pull request #3935 from poolio/rop_bugs

add missing return to GpuFromHost R_op
...@@ -152,7 +152,7 @@ class GpuFromHost(GpuOp): ...@@ -152,7 +152,7 @@ class GpuFromHost(GpuOp):
def R_op(self, inputs, eval_points): def R_op(self, inputs, eval_points):
ev, = eval_points ev, = eval_points
[self(ev)] return [self(ev)]
def infer_shape(self, node, xshp): def infer_shape(self, node, xshp):
return xshp return xshp
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论