mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
compile_model.py -> aoti_custom_class -> torch
The custom command requires `torch` to be installed.
8408522976/test/cpp/aoti_inference/compile_model.py (L1-L7)
Fixes CI failure on trunk:
- https://github.com/pytorch/pytorch/actions/runs/16041370426/job/45275085572#step:22:18348
Pull Request resolved: https://github.com/pytorch/pytorch/pull/157557
Approved by: https://github.com/Skylion007, https://github.com/cyyever