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]: ruff PLC0207 - use maxsplit kwarg (#160107)

Browse Source 
Automatically replaces split with rsplit when relevant and only performs the split up to the first ( or last value). This allows early return of the split function and improve efficiency.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/160107
Approved by: https://github.com/albanD
This commit is contained in:
Aaron Gokaslan
2025-08-08 03:14:55 +00:00
committed by PyTorch MergeBot
parent 3fcd79e023
commit beb4d7816d
15 changed files with 25 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/testing/_internal/common_utils.py
View File

@ -2017,7 +2017,7 @@ def skipIfRocmVersionLessThan(version=None):
def wrap_fn(self, *args, **kwargs):
if TEST_WITH_ROCM:
rocm_version = str(torch.version.hip)
rocm_version = rocm_version.split("-")[0] # ignore git sha
rocm_version = rocm_version.split("-", maxsplit=1)[0] # ignore git sha
rocm_version_tuple = tuple(int(x) for x in rocm_version.split("."))
if rocm_version_tuple is None or version is None or rocm_version_tuple < tuple(version):
reason = f"ROCm {rocm_version_tuple} is available but {version} required"