From 86f053be841023885cb24bef1096229a2e5327df Mon Sep 17 00:00:00 2001
From: Wenkai Du <wenkai.du@amd.com>
Date: Mon, 13 May 2019 23:45:28 +0000
Subject: [PATCH] enable building with mpich

Use following command to build: MPICH=1 make
---
 src/Makefile | 4 ++++
 1 file changed, 4 insertions(+)

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%)