Commit 84e3702e authored by Giovanni Lacopo's avatar Giovanni Lacopo
Browse files

AMD LUMI updates

parent 52ebffe4
Loading
Loading
Loading
Loading
+10 −2
Original line number Diff line number Diff line
@@ -54,14 +54,22 @@ MPICHLIB =
CLANG   = clang
CLANG++ = clang++

HIPCC = hipcc
OPT_HIPCC = -std=c++17 --offload-arch=gfx90a

OPTIMIZE_AMD = -O3 -Ofast -fopenmp -march=native -mavx -mavx2 -fopenmp-targets=amdgcn-amd-amdhsa -Xopenmp-target=amdgcn-amd-amdhsa -march=gfx90a

MPI_INCL = -I/opt/cray/pe/mpich/8.1.27/ofi/crayclang/14.0/include
MPI_LIB = -L/opt/cray/pe/mpich/8.1.27/ofi/crayclang/14.0/lib

RCCL_INCL= -I/opt/rocm-5.2.3/rccl/include
RCCL_LIB= -L/opt/rocm-5.2.3/rccl/lib

HIP_INCL= -I/opt/rocm-5.2.3/hip/include
HIP_LIB= -L/opt/rocm-5.2.3/hip/lib

AMDLIB = $(HIP_INCL) $(HIP_LIB) $(RCCL_INCL) $(RCCL_LIB) -lamdhip64 -lrccl
AMDLIB = $(HIP_INCL) $(HIP_LIB) $(MPI_INCL) $(MPI_LIB) -lamdhip64 -lrccl

AMDLIB_3 = $(HIP_INCL) $(HIP_LIB) $(MPI_INCL) $(MPI_LIB) $(RCCL_INCL) $(RCCL_LIB) -lamdhip64 -lrccl
###########################################################