also set CMAKE_CUDA_COMPILER and CMAKE_CUDA_ARCHITECTURES

bedroge · Dec 2, 2024 · 0857b9f · 0857b9f
1 parent ee21659
commit 0857b9f
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/easybuild/easyblocks/generic/cmakemake.py b/easybuild/easyblocks/generic/cmakemake.py
@@ -328,10 +328,18 @@ def configure_step(self, srcdir=None, builddir=None):
         # ensure CMake uses EB python, not system or virtualenv python
         options.update(get_cmake_python_config_dict())
 
-        # pass the preferred host compiler to the CUDA compiler
+        # pass the preferred host compiler, CUDA compiler, and CUDA architectures to the CUDA compiler
         cuda_root = get_software_root('CUDA')
         if cuda_root:
-            self.cfg.update('preconfigopts', 'CUDAHOSTCXX=%s' % which(os.getenv('CXX', 'g++')))
+            options['CMAKE_CUDA_HOST_COMPILER'] = which(os.getenv('CXX', 'g++'))
+            options['CMAKE_CUDA_COMPILER'] = 'nvcc'
+            cuda_cc = build_option('cuda_compute_capabilities') or self.cfg['cuda_compute_capabilities']
+            if cuda_cc:
+                options['CMAKE_CUDA_ARCHITECTURES'] = '"%s"' % ';'.join([cc.replace('.', '') for cc in cuda_cc])
+            else:
+                raise EasyBuildError('List of CUDA compute capabilities must be specified, either via '
+                                     'cuda_compute_capabilities easyconfig parameter or via '
+                                     '--cuda-compute-capabilities')
 
         if not self.cfg.get('allow_system_boost', False):
             boost_root = get_software_root('Boost')