diff --git a/src/Makefile b/src/Makefile index f1a6a6e2e3..acf41d7e5a 100644 --- a/src/Makefile +++ b/src/Makefile @@ -44,7 +44,11 @@ endif ifeq ($(MPI), 1) HIPCUFLAGS += -DMPI_SUPPORT -I${MPI_HOME}/include HIPLDFLAGS += -L${MPI_HOME}/lib -lmpi +else ifeq($(MPICH), 1) +HIPCUFLAGS += -DMPI_SUPPORT -I/usr/include/mpich +HIPLDFLAGS += -L/usr/lib -lmpi endif + LIBRARIES += rccl HIPLDFLAGS += $(LIBRARIES:%=-l%)