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 ruff PERF fixes to torch (#104917)

Browse Source 
Applies automated ruff fixes in the PERF modules and enables all automatic ones. I also updated ruff which applied some additional fixes.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/104917
Approved by: https://github.com/ezyang, https://github.com/albanD
This commit is contained in:
Aaron Gokaslan
2023-07-11 20:45:18 +00:00
committed by PyTorch MergeBot
parent c9a806be28
commit 2f95a3d0fc
42 changed files with 89 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_hub.py
View File

@ -13,7 +13,7 @@ from torch.testing._internal.common_utils import retry, IS_SANDCASTLE, TestCase
def sum_of_state_dict(state_dict):
s = 0
for _, v in state_dict.items():
for v in state_dict.values():
s += v.sum()
return s