Enable thrust in trilinos-for-albany gpu build, cuda_aware_mpi to gpu…

…_aware_mpi
E3SM-Project · Sep 10, 2024 · 4815633 · 4815633
1 parent abd9ae1
commit 4815633
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/var/spack/repos/builtin/packages/trilinos-for-albany/package.py b/var/spack/repos/builtin/packages/trilinos-for-albany/package.py
@@ -814,11 +814,12 @@ def cmake_args(self):
                 '-DKokkos_ENABLE_CUDA_UVM:BOOL=%s' % 'ON' if '+uvm' in spec else 'OFF',
                 '-DKokkos_ENABLE_IMPL_CUDA_MALLOC_ASYNC:BOOL=OFF',
                 '-DTPL_ENABLE_CUDA:BOOL=ON',
-                '-DTPL_ENABLE_CUSPARSE:BOOL=OFF'
+                '-DTPL_ENABLE_CUSPARSE:BOOL=OFF',
+                '-DTPL_ENABLE_THRUST:BOOL=ON'
             ])
             if '+tpetra' in spec:
                 options.extend([
-                    '-DTpetra_ASSUME_CUDA_AWARE_MPI:BOOL=%s' % 'ON' if '+aware' in spec else 'OFF'
+                    '-DTpetra_ASSUME_GPU_AWARE_MPI:BOOL=%s' % 'ON' if '+aware' in spec else 'OFF'
                 ])
 
         # Fortran lib