frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE][Ez]: FURB148 - remove useless enumerate calls (#145619)

Browse Source 
Remove useless enumerate calls

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145619
Approved by: https://github.com/drisspg
This commit is contained in:
Aaron Gokaslan
2025-01-24 23:37:12 +00:00
committed by PyTorch MergeBot
parent 0741963e01
commit f3304571fc
15 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_jit_fuser_te.py
View File

@ -2499,7 +2499,7 @@ class TestTEFuser(JitTestCase):
for i, func in enumerate(funcs):
num_args = i + 1
for j, gen in enumerate(gen_tensor):
for gen in gen_tensor:
inps = (gen(n), gen(n), gen(n))
func_s = torch.jit.trace(func, inps, check_trace=False)
torch._C._jit_pass_erase_shape_information(func_s.graph)