From ad9aac78dff384e197fe0f1f2f8e704514b3fbdd Mon Sep 17 00:00:00 2001 From: David Addison Date: Mon, 25 Oct 2021 16:30:57 -0700 Subject: [PATCH] Add MPI_IBM build option [ROCm/rccl-tests commit: 7130fa6096466f80b0c310b9a3070b6556c0e158] --- projects/rccl-tests/src/Makefile | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/projects/rccl-tests/src/Makefile b/projects/rccl-tests/src/Makefile index c8491ea537..9a1f62eeb0 100644 --- a/projects/rccl-tests/src/Makefile +++ b/projects/rccl-tests/src/Makefile @@ -64,6 +64,10 @@ ifeq ($(MPI), 1) NVCUFLAGS += -DMPI_SUPPORT -I$(MPI_HOME)/include NVLDFLAGS += -L$(MPI_HOME)/lib -L$(MPI_HOME)/lib64 -lmpi endif +ifeq ($(MPI_IBM),1) +NVCUFLAGS += -DMPI_SUPPORT +NVLDFLAGS += -lmpi_ibm +endif LIBRARIES += curand nccl nvToolsExt NVLDFLAGS += $(LIBRARIES:%=-l%)