[SWDEV-444963] skip test_roi_align (#1379)

pragupta · web-flow · commit 70d92c035853 · 2024-03-21T13:36:22.000-07:00
* [SWDEV-444963] Fix bad import and skip test_roi_align

* Update test_torchinductor.py
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -5197,6 +5197,7 @@ def fn(in_ptr0, in_ptr1, in_ptr2):
             ),
         )
 
+    @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
`@@ -5197,6 +5197,7 @@ def fn(in_ptr0, in_ptr1, in_ptr2):`
`5197`	`5197`	`),`
`5198`	`5198`	`)`
`5199`	`5199`
	`5200`	`+ @skipIfRocm`
`5200`	`5201`	`def test_roi_align(self):`
`5201`	`5202`	`if not has_torchvision_roi_align():`
`5202`	`5203`	`raise unittest.SkipTest("requires torchvision")`