From f1d0bf97252abab5ad36260ec31a4910d2c82b7b Mon Sep 17 00:00:00 2001 From: Jerry Zhang Date: Thu, 2 May 2024 19:37:54 -0700 Subject: [PATCH] Re-enable AOTI tests Summary: att Test Plan: python test/integration/test_integration.py -k test_aoti Reviewers: Subscribers: Tasks: Tags: --- 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 87efb3962a..b7866614f8 100644 --- a/test/integration/test_integration.py +++ b/test/integration/test_integration.py @@ -1432,9 +1432,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")