Adding nvshmem

stacks/pinot-noir-gcc/definitions_cuda.yaml

@@ -5,3 +5,4 @@ definitions:
 
   - cuda_serial_codes:
       - nccl
+      - nvshmem

stacks/pinot-noir-gcc/packages_kuma_h100.yaml

@@ -21,6 +21,10 @@ packages:
     - spec: '+cuda cuda_arch=90 +cuda_uvm +wrapper ~openmptarget'
       when: '%gcc'
 
+  nvshmem:
+    require:
+    - spec: '+cuda cuda_arch=90 +ibgda +nccl +ucx +mpi'
+
   openmpi:
     require:
       - spec: '+cuda cuda_arch=90 target=zen4'

stacks/pinot-noir-gcc/packages_kuma_l40s.yaml

@@ -21,6 +21,10 @@ packages:
     - spec: '+cuda cuda_arch=89 +cuda_uvm +wrapper ~openmptarget'
       when: '%gcc'
 
+  nvshmem:
+    require:
+    - spec: '+cuda cuda_arch=89 +ibgda +nccl +ucx +mpi'
+
   openmpi:
     require:
       - spec: '+cuda cuda_arch=89 target=zen4'