mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
310eb6d8c6
Summary: Fixes https://github.com/pytorch/pytorch/issues/130311. We need to guard CUDA-only code in test_aoti_inference with macros so that it won't fail for CPU-only platform. Pull Request resolved: https://github.com/pytorch/pytorch/pull/134675 Approved by: https://github.com/atalman, https://github.com/chunyuan-w |
||
|---|---|---|
| .. | ||
| aoti_custom_class.cpp | ||
| aoti_custom_class.h | ||
| CMakeLists.txt | ||
| compile_model.py | ||
| test.cpp | ||
| test.py | ||