From a3d3c4db1f9cb97c32933750297b34669b701b55 Mon Sep 17 00:00:00 2001 From: LosCrossos <165311345+loscrossos@users.noreply.github.com> Date: Wed, 28 May 2025 17:39:41 +0200 Subject: [PATCH] adding changes for full CUDA Support for Blackwell. Signed-off-by: LosCrossos <165311345+loscrossos@users.noreply.github.com> --- setup.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/setup.py b/setup.py index f61ca90d..5ad9ac1e 100755 --- a/setup.py +++ b/setup.py @@ -191,6 +191,15 @@ def append_nvcc_threads(nvcc_extra_args): if bare_metal_version >= Version("12.8"): cc_flag.append("-gencode") cc_flag.append("arch=compute_100,code=sm_100") + cc_flag.append("-gencode") + cc_flag.append("arch=compute_101,code=sm_101") + cc_flag.append("-gencode") + cc_flag.append("arch=compute_120,code=sm_120") + if bare_metal_version >= Version("12.9"): + cc_flag.append("-gencode") + cc_flag.append("arch=compute_103,code=sm_103") + cc_flag.append("-gencode") + cc_flag.append("arch=compute_121,code=sm_121") # HACK: The compiler flag -D_GLIBCXX_USE_CXX11_ABI is set to be the same as