OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
a95ceb51a2
pytorch/cmake/Modules_CUDA_fix/upstream
History
Jinzhe Zeng dd2062c737 fix CMake FindCUDA module for cross-compiling (#121590) 
Fix two cross-compiling issues in `FindCUDA.cmake` (xref: https://github.com/conda-forge/pytorch-cpu-feedstock/pull/224).

1. `setup.py` reads the cached `CUDA_TOOLKIT_ROOT_DIR`, so it must be cached.
41286f1505/setup.py (L593)

I also submitted it to the upstream CMake: https://gitlab.kitware.com/cmake/cmake/-/merge_requests/9323.

2. [SBSA toolkit](https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=arm64-sbsa&Compilation=Cross&Distribution=Ubuntu&target_version=20.04&target_type=deb_network_cross) is in `sbsa-linux` directory. See also https://gitlab.kitware.com/cmake/cmake/-/issues/24192

I also submitted it to the upstream CMake: https://gitlab.kitware.com/cmake/cmake/-/merge_requests/9324
Pull Request resolved: https://github.com/pytorch/pytorch/pull/121590
Approved by: https://github.com/malfet
2024-03-11 20:09:52 +00:00
..
FindCUDA remove compute capability 3.5 for CUDA 12 (#114930) 2024-03-06 00:40:57 +00:00
CMakeInitializeConfigs.cmake
FindCUDA.cmake fix CMake FindCUDA module for cross-compiling (#121590) 2024-03-11 20:09:52 +00:00
FindPackageHandleStandardArgs.cmake
FindPackageMessage.cmake
README.md

README.md

If you need to update files under this folder, we recommend you issue PRs against the CMake mainline branch, and then backport it here for earlier CMake compatibility.

See this for more details.