Added functions imports (#1521)

BLOrange-AMD · pruthvistony · commit ad10314ef7ca · 2024-08-11T23:48:13.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
@@ -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
@@ -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
 

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