prgenv-gnu rocm

Author: simonpintarelli

config.yaml

@@ -297,6 +297,13 @@ uenvs:
         eiger: [zen2]
         bristen: [a100]
       develop: True
+    "25.4":
+      recipes:
+        # zen2 and zen3 targets use the same recipe
+        mi300: 25.04/mi300
+      deploy:
+        beverin: [mi300]
+      develop: True
   prgenv-nvfortran:
     "24.11":
       recipes:

recipes/prgenv-gnu/25.4/mi300/compilers.yaml

@@ -0,0 +1,5 @@
+bootstrap:
+  spec: [email protected]
+gcc:
+  specs:
+  - gcc@13

recipes/prgenv-gnu/25.4/mi300/config.yaml

@@ -0,0 +1,9 @@
+name: prgenv-gnu
+spack:
+  #commit: releases/v0.23
+  # 0.24.0.dev0
+  # commit: 19734832eb4eeb6aa15f741846955c41fa82f045
+  commit: releases/v0.23
+  repo: https://github.com/spack/spack.git
+store: /user-environment
+description: GNU Compiler toolchain with cray-mpich, Python, CMake and other development tools for MI300A

recipes/prgenv-gnu/25.4/mi300/environments.yaml

@@ -0,0 +1,53 @@
+gcc-env:
+  compiler:
+      - toolchain: gcc
+        spec: gcc
+  mpi:
+      spec: [email protected]
+      gpu: rocm
+  unify: true
+  specs:
+  - boost +chrono +filesystem +iostreams +mpi +python +regex +serialization +shared +system +timer
+  - cmake
+  - fftw
+  - fmt
+  - gsl
+  - hdf5+cxx+hl+fortran
+  - netlib-scalapack
+  - lua
+  - libtree
+  - lz4
+  - meson
+  - ninja
+  - openblas threads=openmp
+  #- [email protected]
+  - python@3
+  - zlib-ng
+  # add GPU-specific packages here, for easier comparison with mc version
+  - hip ^[email protected] ^llvm@18
+  - hdf5+mpi
+  - hipcub
+  # rocm packages
+  - rocthrust
+  - rocprim
+  - rocblas
+  - rocfft
+  - rocsolver
+  - rocsparse
+  - rocrand
+  - rocm-core
+  - rocm-cmake
+  #- rccl
+  #- aw
+  variants:
+  - +mpi
+  - +rocm
+  - amdgpu_target=gfx942
+  - amdgpu_target_sram_ecc=gfx942
+  views:
+    default:
+      link: roots
+      uenv:
+        add_compilers: true
+        prefix_paths:
+          LD_LIBRARY_PATH: [lib, lib64]

recipes/prgenv-gnu/25.4/mi300/extra/reframe.yaml

@@ -0,0 +1,11 @@
+default:
+  features:
+    - cuda
+    - mpi
+    - osu-micro-benchmarks
+    - openmp
+    - serial
+  cc: mpicc
+  cxx: mpic++
+  ftn: mpifort
+  activation: uenv view develop