提交 31823494 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Forgot grad and R_op for allow_gc.

上级 d0bb8089
...@@ -1882,6 +1882,7 @@ class Scan(PureOp): ...@@ -1882,6 +1882,7 @@ class Scan(PureOp):
else: else:
info['name'] = None info['name'] = None
info['mode'] = self.mode info['mode'] = self.mode
info['allow_gc'] = self.allow_gc
outer_inputs = ([grad_steps] + outer_inputs = ([grad_steps] +
outer_inp_seqs + outer_inp_seqs +
...@@ -2047,6 +2048,7 @@ class Scan(PureOp): ...@@ -2047,6 +2048,7 @@ class Scan(PureOp):
else: else:
info['name'] = None info['name'] = None
info['mode'] = self.mode info['mode'] = self.mode
info['allow_gc'] = self.allow_gc
info['mit_mot_out_slices'] = self.mit_mot_out_slices * 2 info['mit_mot_out_slices'] = self.mit_mot_out_slices * 2
info['destroy_map'] = OrderedDict() info['destroy_map'] = OrderedDict()
new_tap_array = [] new_tap_array = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论