提交 e05720b7 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #2578 from carriepl/revert_scan_bug

Set borrow=False for scan inner inputs
...@@ -543,7 +543,7 @@ class Scan(PureOp): ...@@ -543,7 +543,7 @@ class Scan(PureOp):
self.n_sit_sot + self.n_sit_sot +
self.n_nit_sot) self.n_nit_sot)
wrapped_inputs = [Param(x, borrow=True) for x in self.inputs] wrapped_inputs = [Param(x, borrow=True) for x in self.inputs]
wrapped_outputs = [Out(x, borrow=(x not in self.inputs)) for x in wrapped_outputs = [Out(x, borrow=False) for x in
self.outputs[:slices]] self.outputs[:slices]]
wrapped_outputs += self.outputs[slices:] wrapped_outputs += self.outputs[slices:]
profile = None profile = None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论