提交 b09b776d authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix buf in rval function of make_thunk

上级 abfb6448
...@@ -337,9 +337,9 @@ class ScanOp(PureOp): ...@@ -337,9 +337,9 @@ class ScanOp(PureOp):
# 3.3 construct the rval function # 3.3 construct the rval function
def rval(p=p, i=node_input_storage, o=node_output_storage, n=node): def rval(p=p, i=node_input_storage, o=node_output_storage, n=node):
r = perform(n, [x[0] for x in i], o) r = p(n, [x[0] for x in i], o)
for o in node.outputs: for out in node.outputs:
compute_map[o][0] = True compute_map[out][0] = True
return r return r
rval.inputs = node_input_storage rval.inputs = node_input_storage
rval.outputs = node_output_storage rval.outputs = node_output_storage
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论