pytorch/cmake
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
..
External [BC BREAKING] Remove outdated python submodules (#108236) 2023-09-02 06:24:20 +00:00
Modules added path to correct directory containing headers (#110063) 2023-10-04 21:56:36 +00:00
Modules_CUDA_fix Add 9.0a to cpp_extension supported compute archs (#110587) 2023-10-05 17:41:06 +00:00
public Fix unknown c++ flag detection in CMake (#109000) 2023-09-11 08:32:07 +00:00
Allowlist.cmake
BuildVariables.cmake
Caffe2Config.cmake.in Don't find MKL if it isn't used (#109426) 2023-09-17 03:39:39 +00:00
CheckAbi.cmake
cmake_uninstall.cmake.in
Codegen.cmake
DebugHelper.cmake
Dependencies.cmake [2/N] Clean up CMake target linking (#109986) 2023-10-01 05:36:08 +00:00
FlatBuffers.cmake
GoogleTestPatch.cmake
IncludeSource.cpp.in
iOS.cmake Build and test iOS on GitHub M1 runners (#110406) 2023-10-03 03:17:10 +00:00
Metal.cmake
MiscCheck.cmake
ProtoBuf.cmake
ProtoBufPatch.cmake
Summary.cmake Remove unused build system checks and definitions (#109711) 2023-09-21 16:52:16 +00:00
TorchConfig.cmake.in Revert "[Reland] Upgrade NVTX to NVTX3 (#97582)" 2023-08-15 20:55:12 +00:00
TorchConfigVersion.cmake.in
VulkanCodegen.cmake
VulkanDependencies.cmake