Re-enable AOTI tests (#212)

pytorch · Jun 1, 2024 · 8a4e693 · 8a4e693
1 parent 9dbc554
commit 8a4e693
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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")