提交 7e6c7712 authored 作者: Frederic's avatar Frederic

Skip more useless printing when sub opt don't have print_profile

上级 b13f6ac5
...@@ -1991,23 +1991,23 @@ class EquilibriumOptimizer(NavigatorOptimizer): ...@@ -1991,23 +1991,23 @@ class EquilibriumOptimizer(NavigatorOptimizer):
# Skip opt that have 0 times, they probably wasn't even tried. # Skip opt that have 0 times, they probably wasn't even tried.
print(blanc + " ", ' %.3fs - %s' % (t, o), file=stream) print(blanc + " ", ' %.3fs - %s' % (t, o), file=stream)
print(file=stream) print(file=stream)
if (len(opt.global_optimizers) + len(opt.final_optimizers) == 0 or gf_opts = [o for o in opt.global_optimizers + opt.final_optimizers
# sum([time_opts[o] for o in opt.global_optimizers + opt.final_optimizers]) < 1 or if o.print_profile.func_code is not
False): Optimizer.print_profile.func_code]
if not gf_opts:
return return
print(blanc, "Global and final optimizer", file=stream)
for i in range(len(loop_timing)): for i in range(len(loop_timing)):
print(blanc, "Iter %d" % i, file=stream) print(blanc, "Iter %d" % i, file=stream)
for o, prof in zip(opt.global_optimizers, global_sub_profs[i]): for o, prof in zip(opt.global_optimizers, global_sub_profs[i]):
try: try:
o.print_profile(stream, prof, level + 2) o.print_profile(stream, prof, level + 2)
except NotImplementedError: except NotImplementedError:
import pdb;pdb.set_trace()
print(blanc, "merge not implemented for ", o) print(blanc, "merge not implemented for ", o)
for o, prof in zip(opt.final_optimizers, final_sub_profs[i]): for o, prof in zip(opt.final_optimizers, final_sub_profs[i]):
try: try:
o.print_profile(stream, prof, level + 2) o.print_profile(stream, prof, level + 2)
except NotImplementedError: except NotImplementedError:
import pdb;pdb.set_trace()
print(blanc, "merge not implemented for ", o) print(blanc, "merge not implemented for ", o)
@staticmethod @staticmethod
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论