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

Fix unused Python variables in test/[e-z]* (#136964)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136964
Approved by: https://github.com/justinchuby, https://github.com/albanD
This commit is contained in:
Tom Ritchford
2024-12-18 18:14:52 +00:00
committed by PyTorch MergeBot
parent d298bd840f
commit d8c8ba2440
281 changed files with 508 additions and 565 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_jit_fuser.py
View File

@ -32,7 +32,7 @@ def strip_profiling_nodes(nodes):
def warmup_forward(f, *args):
profiling_count = 2
for i in range(profiling_count):
for _ in range(profiling_count):
results = f(*args)
return results
@ -94,7 +94,7 @@ class TestFuser(JitTestCase):
sin = torch.zeros(0, device="cuda")
cos = torch.zeros(0, device="cuda")
inputs = [sin, cos]
ge = self.checkScript(decode, inputs)
self.checkScript(decode, inputs)
@unittest.skipIf(not RUN_CUDA, "fuser requires CUDA")
def test_arg_configurations_smoke_cuda(self):
@ -587,7 +587,7 @@ class TestFuser(JitTestCase):
return p * (x * x + x)
scripted = torch.jit.script(fn_test_scalar_arg_requires_grad)
out = scripted(x, p)
scripted(x, p)
self.assertAllFused(scripted.graph_for(x, p), except_for=("aten::size", "prim::BroadcastSizes",
"aten::_size_if_not_equal"))