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 38 Packages Projects Releases Wiki Activity
71cb13869b
pytorch/cmake/Modules_CUDA_fix/upstream
History
Dmytro Dzhulgakov a0cea517e7 Add 9.0a to cpp_extension supported compute archs (#110587) 
There's an extended compute capability 9.0a for Hopper that was introduced in Cuda 12.0: https://docs.nvidia.com/cuda/archive/12.0.0/cuda-compiler-driver-nvcc/index.html#gpu-feature-list

E.g. Cutlass leverages it: 5f13dcad78/python/cutlass/emit/pytorch.py (L684)

This adds it to the list of permitted architectures to use in `cpp_extension` directly.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110587
Approved by: https://github.com/ezyang
2023-10-05 17:41:06 +00:00
..
FindCUDA Add 9.0a to cpp_extension supported compute archs (#110587) 2023-10-05 17:41:06 +00:00
CMakeInitializeConfigs.cmake
FindCUDA.cmake [BE] Remove fast_nvcc tool (#96665) 2023-03-14 03:17:31 +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.