提交 6ab9f3dd authored 作者: Razvan Pascanu's avatar Razvan Pascanu

Better names in profile mode for scan

上级 194cb5d4
...@@ -1141,6 +1141,7 @@ class Scan(Op): ...@@ -1141,6 +1141,7 @@ class Scan(Op):
# scan does is a constant of 1, -1 or 0 then we can remove scan # scan does is a constant of 1, -1 or 0 then we can remove scan
# from the graph # from the graph
self.mode = mode self.mode = mode
self.name = name
self.truncate_gradient = truncate_gradient self.truncate_gradient = truncate_gradient
self.go_backwards = go_backwards self.go_backwards = go_backwards
self.slice_to_seqs = slice_to_seqs self.slice_to_seqs = slice_to_seqs
...@@ -1157,6 +1158,9 @@ class Scan(Op): ...@@ -1157,6 +1158,9 @@ class Scan(Op):
linker=mode_instance.provided_linker) linker=mode_instance.provided_linker)
compile.profilemode.prof_mode_instance_to_print.append(mode_instance) compile.profilemode.prof_mode_instance_to_print.append(mode_instance)
self.mode_instance = mode_instance self.mode_instance = mode_instance
if self.name:
self.mode_instance.message=self.name+" sub profile"
else:
self.mode_instance.message="Scan sub profile" self.mode_instance.message="Scan sub profile"
if name is None: name = 'scan_fn' if name is None: name = 'scan_fn'
...@@ -1167,7 +1171,11 @@ class Scan(Op): ...@@ -1167,7 +1171,11 @@ class Scan(Op):
assert not numpy.any([isinstance(x.variable,SharedVariable) for x in assert not numpy.any([isinstance(x.variable,SharedVariable) for x in
self.fn.maker.inputs]) self.fn.maker.inputs])
def __str__(self):
if self.name:
return self.name
else:
return 'scan'
def make_node(self,*inputs): def make_node(self,*inputs):
assert all(isinstance(i, gof.Variable) for i in inputs) assert all(isinstance(i, gof.Variable) for i in inputs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论