提交 530f513b authored 作者: Razvan Pascanu's avatar Razvan Pascanu

Fixed profiling with scan.

上级 9c4c58ad
......@@ -610,11 +610,13 @@ class Scan(Op):
t_call = time.time() - t0_call
if hasattr(self.fn.maker.mode,'fct_call_time'):
self.fn.maker.mode.fct_call_time[self.fn] += t_call
self.fn.maker.mode.fct_call_time[self.fn] += t_fn
self.fn.maker.mode.fct_call[self.fn] += n_steps
self.fn.maker.mode.call_time += t_call
self.fn.maker.mode.call_time += t_fn
self.fn.maker.mode.fn_time += t_fn
self.t_call = t_call
self.t_fn = t_fn
### Infer Shape
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论