From 69a0fbd2cbb0774f832b8a37a07542ef57991dbf Mon Sep 17 00:00:00 2001 From: Matt Larsen Date: Fri, 1 May 2020 01:07:12 -0700 Subject: raja: add awareness of the cuda architecture (#16377) --- var/spack/repos/builtin/packages/raja/package.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/var/spack/repos/builtin/packages/raja/package.py b/var/spack/repos/builtin/packages/raja/package.py index ccb1863444..10a9265257 100644 --- a/var/spack/repos/builtin/packages/raja/package.py +++ b/var/spack/repos/builtin/packages/raja/package.py @@ -6,7 +6,7 @@ from spack import * -class Raja(CMakePackage): +class Raja(CMakePackage, CudaPackage): """RAJA Parallel Framework.""" homepage = "http://software.llnl.gov/RAJA/" @@ -47,4 +47,9 @@ class Raja(CMakePackage): '-DENABLE_CUDA=On', '-DCUDA_TOOLKIT_ROOT_DIR=%s' % (spec['cuda'].prefix)]) + cuda_value = spec.variants['cuda_arch'].value + cuda_arch = cuda_value[0] + if cuda_arch is not None: + options.append('-DCUDA_ARCH=sm_{0}'.format(cuda_arch)) + return options -- cgit v1.2.3-70-g09d2