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

Remove unused Python variables in torch/[b-z]* (#136963)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136963
Approved by: https://github.com/ezyang
This commit is contained in:
Tom Ritchford
2024-10-19 13:25:28 +00:00
committed by PyTorch MergeBot
parent fb44658415
commit c0582fd0f8
152 changed files with 376 additions and 528 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torch/sparse/_triton_ops_meta.py
View File

@ -379,7 +379,6 @@ def minimize(
minimizer_key = (
initial_key if initial_key in minimizer_keys else min(minimizer_keys)
)
minimizer_target = all_values[minimizer_key]
parameters = from_key(minimizer_key, parameters)
speedup_incr = (1 - minimal_target / reference_target) * 100
if speedup_incr < 0:
@ -555,7 +554,7 @@ def optimize_scatter_mm(
return value
return next_value
meta, speedup, timing, sensitivity_message = minimize(
meta, speedup, timing, _sensitivity_message = minimize(
bench, initial_meta, reference_meta, step_meta_parameter
)
if initial_meta is not reference_meta and initial_meta == meta and not force:
@ -833,7 +832,7 @@ def main(op="scatter_mm", force=False, dtype=torch.float16, verbose=True):
raise NotImplementedError(op)
except KeyboardInterrupt:
break
except Exception as msg:
except Exception:
dump()
raise
dump()