diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/raja/package.py | 7 |
1 files changed, 6 insertions, 1 deletions
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 |