cpp_extension.py
13.4 KB
-
Add support for building Ampere GPU, CUDA11 and CUDNN8 · c40eaf7bSummary: This commit fixes the issue on building with CUDA11 and CUDNN8. Besides, C++14 is enabled by default instead of C++11 to support CUB 1.9+, and for this reason, the compiler is required to be gcc5/clang6/msvc141 or higher.
Ting PAN committed