From eaeb6467422c88d90283060cabdd5c4f5032deb4 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Thu, 11 Apr 2024 11:36:21 -0500 Subject: [PATCH] Skip tests that crash on MI210 for now. --- models/turbine_models/tests/sdxl_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/models/turbine_models/tests/sdxl_test.py b/models/turbine_models/tests/sdxl_test.py index 387eabc87..a45fd7ca4 100644 --- a/models/turbine_models/tests/sdxl_test.py +++ b/models/turbine_models/tests/sdxl_test.py @@ -93,7 +93,7 @@ def setUp(self): ) def test01_ExportClipModels(self): - if arguments["device"] in ["vulkan", "cuda"]: + if arguments["device"] in ["vulkan", "cuda", "rocm"]: self.skipTest( "Compilation error on vulkan; Runtime error on rocm; To be tested on cuda." ) @@ -215,7 +215,7 @@ def test01_ExportClipModels(self): np.testing.assert_allclose(torch_output_2, turbine_2[0], rtol, atol) def test02_ExportUnetModel(self): - if arguments["device"] in ["vulkan", "cuda"]: + if arguments["device"] in ["vulkan", "cuda", "rocm"]: self.skipTest( "Unknown error on vulkan; Runtime error on rocm; To be tested on cuda." ) @@ -325,7 +325,7 @@ def test02_ExportUnetModel(self): np.testing.assert_allclose(torch_output, turbine, rtol, atol) def test03_ExportVaeModelDecode(self): - if arguments["device"] in ["vulkan", "cuda"]: + if arguments["device"] in ["vulkan", "cuda", "rocm"]: self.skipTest( "Compilation error on vulkan; Runtime error on rocm; To be tested on cuda." )