From 32ffc3529655735d1c04dafb006508a3b01b364b Mon Sep 17 00:00:00 2001 From: blorange-amd Date: Fri, 2 Aug 2024 15:45:12 +0000 Subject: [PATCH] Added functions imports --- test/inductor/test_torchinductor.py | 1 + test/inductor/test_torchinductor_dynamic_shapes.py | 1 + 2 files changed, 2 insertions(+) diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py index 5356a5d5b15e18..f41a3c5c517e9e 100644 --- a/test/inductor/test_torchinductor.py +++ b/test/inductor/test_torchinductor.py @@ -71,6 +71,7 @@ from torch.testing._internal.common_device_type import ( _has_sufficient_memory, expectedFailureXPU, + get_desired_device_type_test_bases, ) from torch.testing._internal.common_dtype import all_types, get_all_dtypes from torch.testing._internal.common_utils import ( diff --git a/test/inductor/test_torchinductor_dynamic_shapes.py b/test/inductor/test_torchinductor_dynamic_shapes.py index 4c143b0e4c7713..ef333bb06c251c 100644 --- a/test/inductor/test_torchinductor_dynamic_shapes.py +++ b/test/inductor/test_torchinductor_dynamic_shapes.py @@ -31,6 +31,7 @@ TEST_CUDA_MEM_LEAK_CHECK, TEST_WITH_ASAN, TEST_WITH_ROCM, + skipIfRocm, ) from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_CPU, HAS_GPU