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"