mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Summary: To fix https://github.com/pytorch/pytorch/issues/159400. Currently, test_aoti_abi_check and test_aoti_inference need to be built in two passes, first build pytorch using the regular `pythonsetup.py develop` and then build with `CMAKE_FRESH=1 BUILD_AOT_INDUCTOR_TEST=1 python setup.py devleop`. This is cumbersome. Fix by rewriting CMakeLists.txt for test_aoti_inference to one-pass build which runs AOTI to compile models at the test time. Also update CI test script to get rid of two-pass build. For test_aoti_abi_check, it is not AOTI specific, so we make it not guarded by BUILD_AOT_INDUCTOR_TEST. Pull Request resolved: https://github.com/pytorch/pytorch/pull/164277 Approved by: https://github.com/janeyx99 |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| main.cpp | ||
| README.md | ||
| test_cast.cpp | ||
| test_devicetype.cpp | ||
| test_dtype.cpp | ||
| test_exception.cpp | ||
| test_macros.cpp | ||
| test_math.cpp | ||
| test_rand.cpp | ||
| test_scalartype.cpp | ||
| test_vec_half.cpp | ||
| test_vec.cpp | ||
Tests in this directory are meant to guard certain ATen/c10 util functions and data structures are implemented in a header-only fashion, to make sure AOTInductor generated CPU model code is ABI backward-compatible.