未验证 提交 d8ed3ce4 编写于 作者: T TongXU 提交者: GitHub

remove the print line in _einsum_flops_compute function (#1885)

上级 c4873722
......@@ -734,7 +734,6 @@ def _einsum_flops_compute(equation, *operands):
np_arrs = [np.zeros(s) for s in input_shapes]
optim = np.einsum_path(equation, *np_arrs, optimize="optimal")[1]
for line in optim.split("\n"):
print(line.lower())
if "optimized flop" in line.lower():
flop = int(float(line.split(":")[-1]))
return flop, 0
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册