From 07b8876277596948512fab4df2b705d6e1e6ca15 Mon Sep 17 00:00:00 2001
From: Sourav Chakraborty <souchakr@amd.com>
Date: Fri, 10 Apr 2020 07:51:39 -0700
Subject: [PATCH] Improve makefile to avoid LD_LIBRARY_PATH

---
 src/Makefile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Makefile b/src/Makefile
index e109b8c3ae..8a63340968 100644
--- a/src/Makefile
+++ b/src/Makefile
@@ -37,11 +37,11 @@ endif
 BUILDDIR ?= ../build
 ifneq ($(NCCL_HOME), "")
 HIPCUFLAGS += -I$(NCCL_HOME)/include/
-HIPLDFLAGS   += -L$(NCCL_HOME)/lib
+HIPLDFLAGS   += -Wl,-rpath,$(NCCL_HOME)/lib -L$(NCCL_HOME)/lib
 endif
 
 ifeq ($(MPI), 1)
-HIPCUFLAGS += -DMPI_SUPPORT -I${MPI_HOME}/include
+HIPCUFLAGS += -DMPI_SUPPORT -I${MPI_HOME}/include -I${MPI_HOME}/include/mpi
 HIPLDFLAGS += -L${MPI_HOME}/lib -lmpi
 else ifeq ($(MPICH), 1)
 HIPCUFLAGS += -DMPI_SUPPORT -I/usr/include/mpich