packages.yaml

packages:
  all:
    providers:
      mpi: [openmpi, intel-oneapi-mpi]
      jpeg: [libjpeg]
    require:
      - spec: ^intel-oneapi-mpi
        when: '%oneapi ^mpi'

      - spec: ^openmpi
        when: '%gcc ^mpi'

      - spec: ^openmpi
        when: '%aocc ^mpi'

      - spec: ^nvhpc+mpi
        when: '%nvhpc ^mpi'

      - spec: "^cuda%gcc@13.2.0"
        when: '%gcc@13.2.0 ^cuda'

      - spec: "^cuda%oneapi"
        when: '%oneapi ^cuda'

      - spec: "+ipo"
        when: '%gcc'

  # ----------------------------------------------------------------------------
  # Requirements
  # ----------------------------------------------------------------------------
  cuda:
    require:
    - spec: '@12.4.0'

  hdf5:
    variants: +hl +threadsafe +szip +cxx +fortran

  libfabric:
    require:
    - spec: fabrics=mlx,mrail,psm3,verbs,udp,tcp,sockets,shm

  openmpi:
    require:
    - spec: '@5.0.3 fabrics=ofi,ucx,verbs schedulers=slurm ~rsh +romio romio-filesystem=gpfs'
    - spec: ^ucx +cuda
      when: +cuda

  python:
    require:
    - spec: +optimizations
      when: '%gcc'

  ucx:
    require:
    - spec: +rdmacm +rc +dc +ud +cma +verbs
    - spec: +gdrcopy
      when: +cuda

  # ----------------------------------------------------------------------------
  # System dependencies
  # ----------------------------------------------------------------------------
  rdma-core:
    buildable: false
    externals:
    - spec: 'rdma-core@44.1'
      prefix: /usr
    require:
    - spec: '%gcc@11.4.1 arch=linux-rhel9-x86_64_v2'

  slurm: