nvhpc.nu

const nvhome = "/opt/nvidia/hpc_sdk"
const target = "Linux_x86_64"
const version = "2024"
const nvcudadir = "/opt/nvidia/hpc_sdk/Linux_x86_64/2024/cuda"
const nvcompdir = "/opt/nvidia/hpc_sdk/Linux_x86_64/2024/compilers"
const nvmathdir = "/opt/nvidia/hpc_sdk/Linux_x86_64/2024/math_libs"
const nvcommdir = "/opt/nvidia/hpc_sdk/Linux_x86_64/2024/comm_libs"

export-env {
  load-env {
    NVHPC: $nvhome,
    NVHPC_ROOT: $"($nvhome)/($target)/($version)",
    CC: $"($nvcompdir)/bin/nvc",
    CXX: $"($nvcompdir)/bin/nvc++",
    FC: $"($nvcompdir)/bin/nvfortran",
    F90: $"($nvcompdir)/bin/nvfortran",
    F77: $"($nvcompdir)/bin/nvfortran",
    CPP: "cpp",
    MANPATH: $"($nvcompdir)/man",
    OPAL_PREFIX: $"($nvcommdir)/mpi",
    PATH: ( 
      $env |
      get -i PATH |
      prepend [
        $"($nvcudadir)/bin"
        $"($nvcompdir)/bin"
        $"($nvcommdir)/mpi/bin"
        $"($nvcompdir)/extras/qd/bin"
      ]  
    )
    LD_LIBRARY_PATH: (
      $env |
      get -i LD_LIBRARY_PATH |
      prepend [
        $"($nvcudadir)/lib64"
        $"($nvcudadir)/extras/CUPTI/lib64"
        $"($nvcompdir)/extras/qd/lib"
        $"($nvcompdir)/lib"
        $"($nvmathdir)/lib64"
        $"($nvcommdir)/mpi/lib"
        $"($nvcommdir)/nccl/lib"
        $"($nvcommdir)/nvshmem/lib"
      ]
    )
    CPATH: (
      $env |
      get -i CPATH |
      prepend [
        $"($nvmathdir)/include"
        $"($nvcommdir)/mpi/include"
        $"($nvcommdir)/nccl/include"
        $"($nvcommdir)/nvshmem/include"
        $"($nvcompdir)/extras/qd/include/qd"
      ]
    )
  }
}