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]: Apply PERF401 autofixes from ruff (#140980)

Browse Source 
* Automatically applies ruff rule 401. Turns loops into equivalent list comprehensions which are faster and do not leak the scope of the loop variables.
* list comprehensions not only often have better typing, but are 50+% faster than for loops on overhead. They also preserve length information etc and are better for the interpreter to optimize.
* Manually went back and made mypy happy after the change.
* Also fixed style lints in files covered by flake8 but not by pyfmt

Pull Request resolved: https://github.com/pytorch/pytorch/pull/140980
Approved by: https://github.com/justinchuby, https://github.com/malfet
This commit is contained in:
Aaron Gokaslan
2024-11-20 17:52:07 +00:00
committed by PyTorch MergeBot
parent 8d708090c0
commit 12e95aa4ee
133 changed files with 611 additions and 761 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/sparse/_triton_ops_meta.py
View File

@ -233,8 +233,7 @@ def dump():
for op_key in sorted(_operation_device_version_data, key=sort_key):
data_part.append(" " + repr(op_key).replace("'", '"') + ": {")
op_data = _operation_device_version_data[op_key]
for key in sorted(op_data):
data_part.append(f" {key}: {op_data[key]},")
data_part.extend(f" {key}: {op_data[key]}," for key in sorted(op_data))
data_part.append(" },")
new_content = part1 + "\n".join(data_part) + "\n" + part2
if current_content != new_content: