From fae031751e862ea54b3ad14affa0b5160065b577 Mon Sep 17 00:00:00 2001 From: Dmitry Nikolaev Date: Wed, 1 Oct 2025 18:22:40 +0000 Subject: [PATCH] lint --- test/test_linalg.py | 2 +- torch/testing/_internal/common_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_linalg.py b/test/test_linalg.py index e4f33578de340..bc5f14a5a225c 100644 --- a/test/test_linalg.py +++ b/test/test_linalg.py @@ -23,7 +23,7 @@ TEST_WITH_ROCM, IS_FBCODE, IS_REMOTE_GPU, iter_indices, make_fullrank_matrices_with_distinct_singular_values, freeze_rng_state, IS_ARM64, IS_SANDCASTLE, TEST_OPT_EINSUM, parametrize, skipIfTorchDynamo, - setBlasBackendsToDefaultFinally, setLinalgBackendsToDefaultFinally, serialTest, + skipIfRocmArch, setBlasBackendsToDefaultFinally, setLinalgBackendsToDefaultFinally, serialTest, runOnRocmArch, MI300_ARCH, NAVI_ARCH, TEST_CUDA) from torch.testing._internal.common_device_type import \ (instantiate_device_type_tests, dtypes, has_cusolver, has_hipsolver, diff --git a/torch/testing/_internal/common_utils.py b/torch/testing/_internal/common_utils.py index c8f2060ebab37..a9c5f2911d39f 100644 --- a/torch/testing/_internal/common_utils.py +++ b/torch/testing/_internal/common_utils.py @@ -2017,7 +2017,7 @@ def getRocmVersion() -> tuple[int, int, int]: rocm_version = str(torch.version.hip) rocm_version = rocm_version.split("-")[0] # ignore git sha rocm_version_tuple = tuple(int(x) for x in rocm_version.split(".")) - return rocm_version_tuple + return (rocm_version_tuple[0], rocm_version_tuple[1], rocm_version_tuple[2]) # Skips a test on CUDA if ROCm is available and its version is lower than requested. def skipIfRocmVersionLessThan(version=None):