Bad import in test_torchinductor and skip torchvision related UT (#1374)

pragupta · pruthvistony · commit 1e0c5c4f6019 · 2024-08-12T00:01:11.000-05:00
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -86,6 +86,7 @@
     skipIfXpu,
     subtest,
     skipIfRocmArch,
+    subtest,
     TEST_WITH_ASAN,
     TEST_WITH_ROCM,
 )
@@ -6823,6 +6824,7 @@ def fn(in_ptr0, in_ptr1, in_ptr2):
         )
 
     @skipIfWindows
+    @skipIfRocm
     def test_roi_align(self):
         if not has_torchvision_roi_align():
             raise unittest.SkipTest("requires torchvision")

Original file line number	Diff line number	Diff line change
`@@ -86,6 +86,7 @@`
`86`	`86`	`skipIfXpu,`
`87`	`87`	`subtest,`
`88`	`88`	`skipIfRocmArch,`
	`89`	`+ subtest,`
`89`	`90`	`TEST_WITH_ASAN,`
`90`	`91`	`TEST_WITH_ROCM,`
`91`	`92`	`)`
`@@ -6823,6 +6824,7 @@ def fn(in_ptr0, in_ptr1, in_ptr2):`
`6823`	`6824`	`)`
`6824`	`6825`
`6825`	`6826`	`@skipIfWindows`
	`6827`	`+ @skipIfRocm`
`6826`	`6828`	`def test_roi_align(self):`
`6827`	`6829`	`if not has_torchvision_roi_align():`
`6828`	`6830`	`raise unittest.SkipTest("requires torchvision")`