From 8a4e6938dbf52011483a0a22deb7254d4e9a29bc Mon Sep 17 00:00:00 2001 From: Jerry Zhang Date: Fri, 31 May 2024 22:23:06 -0700 Subject: [PATCH] Re-enable AOTI tests (#212) --- test/integration/test_integration.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/test/integration/test_integration.py b/test/integration/test_integration.py index 2cd34f427..74a163ee1 100644 --- a/test/integration/test_integration.py +++ b/test/integration/test_integration.py @@ -1520,9 +1520,7 @@ def test_aoti(self, api, test_device, test_dtype): if not TORCH_VERSION_AFTER_2_4: self.skipTest("aoti compatibility requires 2.4+.") - if test_device == "cuda": - self.skipTest("AOTI has some issues in cuda test right now, skipping") - + print(f"TestAOTI: {api}, {test_device}, {test_dtype}") logger.info(f"TestAOTI: {api}, {test_device}, {test_dtype}") if api is change_linear_weights_to_int8_dqtensors and test_device == "cuda": self.skipTest(f"{api} in {test_device} is not support for aoti compilation yet")