Added functions imports (#1521)

BLOrange-AMD · pruthvistony · commit 87728a1eb173 · 2024-08-12T00:21:28.000-05:00
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
@@ -33,6 +33,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
`@@ -33,6 +33,7 @@`
`33`	`33`	`TEST_CUDA_MEM_LEAK_CHECK,`
`34`	`34`	`TEST_WITH_ASAN,`
`35`	`35`	`TEST_WITH_ROCM,`
	`36`	`+ skipIfRocm,`
`36`	`37`	`)`
`37`	`38`	`from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_CPU, HAS_GPU`
`38`	`39`