Skip to content

Commit 8a4e693

Browse files
authored
Re-enable AOTI tests (#212)
1 parent 9dbc554 commit 8a4e693

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

test/integration/test_integration.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1520,9 +1520,7 @@ def test_aoti(self, api, test_device, test_dtype):
15201520
if not TORCH_VERSION_AFTER_2_4:
15211521
self.skipTest("aoti compatibility requires 2.4+.")
15221522

1523-
if test_device == "cuda":
1524-
self.skipTest("AOTI has some issues in cuda test right now, skipping")
1525-
1523+
print(f"TestAOTI: {api}, {test_device}, {test_dtype}")
15261524
logger.info(f"TestAOTI: {api}, {test_device}, {test_dtype}")
15271525
if api is change_linear_weights_to_int8_dqtensors and test_device == "cuda":
15281526
self.skipTest(f"{api} in {test_device} is not support for aoti compilation yet")

0 commit comments

Comments
 (0)