Lammps update to separate lammps + gpu and lammps + kokkos

docs/uenv-lammps.md

@@ -4,22 +4,30 @@
 
 ## Running
 
+We provide two versions of LAMMPS, one with the kokkos package enabled, and one with the GPU packaged enabled. These can be loaded as follows:
 ```bash
 uenv start <LAMMPS_UENV>
-uenv view lammps
+uenv view lammps:kokkos
+```
+
+or
+
+```bash
+uenv start <LAMMPS_UENV>
+uenv view lammps:gpu
 ```
 
 !!! warning
     [LAMMPS] is built with GPU-aware MPI. Make sure to set `MPICH_GPU_SUPPORT_ENABLED=1` when running [LAMMPS].
 
 ## Building from source
 
-The [LAMMPS] `uenv` provides all the dependencies required to build [LAMMPS] from source, including kokkos. You can follow these steps to build [LAMMPS] from source:
+The [LAMMPS] `uenv` provides all the dependencies required to build [LAMMPS] from source. You can follow these steps to build [LAMMPS] from source:
 
 ```bash
 # Start uenv and load develop view
 uenv start <LAMMPS_UENV>
-uenv view develop
+uenv view lammps:develop-kokkos # or uenv view lammps:develop-gpu, if building using the GPU package
 
 # cd to LAMMPS source directory
 cd <PATH_TO_LAMMPS_SOURCE>

recipes/lammps/2024/gh200/config.yaml

@@ -4,4 +4,4 @@ description: lammps build targetting GH200
 spack:
   repo: https://github.com/spack/spack.git
   commit: v0.22.0
-modules: true
+modules: false

recipes/lammps/2024/gh200/environments.yaml

@@ -1,4 +1,4 @@
-lammps:
+lammps-kokkos:
   compiler:
   - toolchain: gcc
     spec: [email protected]
@@ -11,16 +11,56 @@ lammps:
   - cmake
   - cuda@12
   - [email protected] +wrapper +cuda cuda_arch=90
-  - mpi
-  - cray-mpich
-  - [email protected] +cuda cuda_arch=90 +python +kokkos
+  # lammps + kokkos package
+  - [email protected] +kokkos +cuda cuda_arch=90 +python +extra-dump +plumed +replica
   variants:
   - +mpi
   - +cuda
   - cuda_arch=90
   views:
-    develop:
-      link: roots
+    develop-kokkos:
+      link: all
       exclude: ["lammps"]
-    lammps:
-      link: roots
+      uenv:
+        add_compilers: true
+        prefix_paths:
+          LD_LIBRARY_PATH: [lib, lib64]
+    kokkos:
+      link: all
+      uenv:
+        add_compilers: true
+        prefix_paths:
+          LD_LIBRARY_PATH: [lib, lib64]
+lammps-gpu:
+  compiler:
+  - toolchain: gcc
+    spec: [email protected]
+  mpi:
+    spec: cray-mpich
+    gpu: cuda
+  unify: true
+  specs:
+  - [email protected]
+  - cmake
+  - cuda@12
+  # lammps + GPU package, MPS enabled
+  - [email protected] ~kokkos +cuda cuda_arch=90 +python +extra-dump +cuda_mps +plumed +replica
+  variants:
+  - +mpi
+  - +cuda
+  - cuda_arch=90
+  views:
+    develop-gpu:
+      link: all
+      exclude: ["lammps"]
+      uenv:
+        add_compilers: true
+        prefix_paths:
+          LD_LIBRARY_PATH: [lib, lib64]
+    gpu:
+      link: all
+      uenv:
+        add_compilers: true
+        prefix_paths:
+          LD_LIBRARY_PATH: [lib, lib64]
+