From 8c303ab4e7aa59fccc9e9031ec50ce75fb99ea43 Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Tue, 9 Apr 2024 21:22:11 +0000 Subject: [PATCH] Add test_aot_inductor to test_inductor (#123340) AOTI changes have been breaking for ROCm on trunk because we do not have testing of AOTI in inductor/pull/trunk workflow for ROCm. This PR adds `test_aot_inductor` to inductor workflow to catch such issues. More context here: https://github.com/pytorch/pytorch/pull/123164#issuecomment-2033494012 Runtime increase for inductor workflow: CUDA: PR corresponding to base commit used for this PR: [100 mins](https://github.com/pytorch/pytorch/actions/runs/8545475047/job/23415210028?pr=123290) This PR: [183 mins](https://github.com/pytorch/pytorch/actions/runs/8562003098/job/23465530389?pr=123340) ROCM: PR corresponding to base commit used for this PR: [105 mins](https://github.com/pytorch/pytorch/actions/runs/8545475047/job/23416422145?pr=123290) This PR: [148 mins](https://github.com/pytorch/pytorch/actions/runs/8562003098/job/23466516866?pr=123340) Pull Request resolved: https://github.com/pytorch/pytorch/pull/123340 Approved by: https://github.com/atalman, https://github.com/desertfire --- .ci/pytorch/test.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.ci/pytorch/test.sh b/.ci/pytorch/test.sh index c10a4450f625..48ea7d742ba6 100755 --- a/.ci/pytorch/test.sh +++ b/.ci/pytorch/test.sh @@ -327,7 +327,7 @@ test_inductor() { python tools/dynamo/verify_dynamo.py python test/run_test.py --inductor --include test_modules test_ops test_ops_gradients test_torch --verbose # Do not add --inductor for the following inductor unit tests, otherwise we will fail because of nested dynamo state - python test/run_test.py --include inductor/test_torchinductor inductor/test_torchinductor_opinfo --verbose + python test/run_test.py --include inductor/test_torchinductor inductor/test_torchinductor_opinfo inductor/test_aot_inductor --verbose # docker build uses bdist_wheel which does not work with test_aot_inductor # TODO: need a faster way to build