Added functions imports (#1521)

BLOrange-AMD · dnikolaev-amd · commit 2ec017267f35 · 2024-09-16T16:06:12.000Z
Fixes
inductor.test_torchinductor_dynamic_shapes::TestInductorDynamicCUDA::test_item_unbacked_stride_nobreak_cuda
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -70,6 +70,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
@@ -32,6 +32,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
 

Original file line number	Diff line number	Diff line change
`@@ -70,6 +70,7 @@`
`70`	`70`	`from torch.testing._internal.common_device_type import (`
`71`	`71`	`_has_sufficient_memory,`
`72`	`72`	`expectedFailureXPU,`
	`73`	`+ get_desired_device_type_test_bases,`
`73`	`74`	`)`
`74`	`75`	`from torch.testing._internal.common_dtype import all_types, get_all_dtypes`
`75`	`76`	`from torch.testing._internal.common_utils import (`
Original file line number	Diff line number	Diff line change
`@@ -32,6 +32,7 @@`
`32`	`32`	`TEST_CUDA_MEM_LEAK_CHECK,`
`33`	`33`	`TEST_WITH_ASAN,`
`34`	`34`	`TEST_WITH_ROCM,`
	`35`	`+ skipIfRocm,`
`35`	`36`	`)`
`36`	`37`	`from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_CPU, HAS_GPU`
`37`	`38`