提交 b2aa906d authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Set borrow=False for scan inner inputs

上级 cb4ff4b0
...@@ -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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论