From f613e10f247eb3f1261f76f97787708501b39d08 Mon Sep 17 00:00:00 2001 From: psakievich Date: Thu, 19 Nov 2020 17:24:42 -0700 Subject: Trilinos: Add CUDA relocatable code flag (#19993) * Add relocatable code flag to trilinos * Make CUDA RDC and varainat * adjust default of cuda_rdc --- var/spack/repos/builtin/packages/trilinos/package.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/trilinos/package.py b/var/spack/repos/builtin/packages/trilinos/package.py index 2d341b245a..50227fec2a 100644 --- a/var/spack/repos/builtin/packages/trilinos/package.py +++ b/var/spack/repos/builtin/packages/trilinos/package.py @@ -77,6 +77,8 @@ class Trilinos(CMakePackage, CudaPackage): description='Compile with Fortran support') variant('wrapper', default=False, description="Use nvcc-wrapper for CUDA build") + variant('cuda_rdc', default=False, + description='turn on RDC for CUDA build') variant('cxxstd', default='11', values=['11', '14', '17'], multi=False) variant('hwloc', default=False, description='Enable hwloc') @@ -328,6 +330,7 @@ class Trilinos(CMakePackage, CudaPackage): conflicts('+adios2', when='@:12.14.1') conflicts('+adios2', when='@xsdk-0.2.0') conflicts('+pnetcdf', when='~netcdf') + conflicts('+cuda_rdc', when='~cuda') conflicts('+wrapper', when='~cuda') conflicts('+wrapper', when='%clang') conflicts('cxxstd=11', when='+wrapper ^cuda@6.5.14') @@ -752,6 +755,10 @@ class Trilinos(CMakePackage, CudaPackage): define('Kokkos_ENABLE_CUDA', True), define('Kokkos_ENABLE_CUDA_UVM', True), define('Kokkos_ENABLE_CUDA_LAMBDA', True)]) + if '+cuda_rdc' in spec: + options.append(define( + 'Kokkos_ENABLE_CUDA_RELOCATABLE_DEVICE_CODE', + True)) for iArchCC in spec.variants['cuda_arch'].value: options.append(define( "Kokkos_ARCH_" + -- cgit v1.2.3-60-g2f50