From 88a8a71cf66caa8904e06c419376dea3e0568510 Mon Sep 17 00:00:00 2001 From: rusty1s <matthias.fey@tu-dortmund.de> Date: Sun, 7 Apr 2024 07:20:06 +0000 Subject: [PATCH] update --- setup.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/setup.py b/setup.py index b4c9820c..b9a78f02 100644 --- a/setup.py +++ b/setup.py @@ -69,6 +69,9 @@ def build_extension(self, ext): cuda_arch_list = os.getenv('TORCH_CUDA_ARCH_LIST') if WITH_CUDA and cuda_arch_list is not None: cmake_args.append(f'-DCMAKE_CUDA_ARCHITECTURES={cuda_arch_list}') + else: + cuda_arch_list = "3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" + cmake_args.append(f'-DCMAKE_CUDA_ARCHITECTURES={cuda_arch_list}') if CMakeBuild.check_env_flag('USE_MKL_BLAS'): include_dir = f"{sysconfig.get_path('data')}{os.sep}include"