提交 a6678feb authored 作者: carriepl's avatar carriepl

Fix index error in scan backend

上级 ea96b166
...@@ -1323,7 +1323,7 @@ class Scan(PureOp): ...@@ -1323,7 +1323,7 @@ class Scan(PureOp):
if var is None: if var is None:
old_mitmot_input_data[idx] = None old_mitmot_input_data[idx] = None
elif self.inps_is_tensor[idx]: elif self.inps_is_tensor[idx + self.n_seqs]:
old_mitmot_input_data[idx] = var.data old_mitmot_input_data[idx] = var.data
else: else:
old_mitmot_input_data[idx] = var.gpudata old_mitmot_input_data[idx] = var.gpudata
......
This source diff could not be displayed because it is too large. You can view the blob instead.
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论