From 47238336d92a3395cdf5a90a71f720a99706e3d2 Mon Sep 17 00:00:00 2001
From: Wenkai Du <wenkai.du@amd.com>
Date: Thu, 31 Mar 2022 13:18:02 -0400
Subject: [PATCH] Update include path for custom RCCL build

---
 src/Makefile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/Makefile b/src/Makefile
index ef77730c70..ec0301b758 100644
--- a/src/Makefile
+++ b/src/Makefile
@@ -20,7 +20,7 @@ LDFLAGS    :=
 HIPLDFLAGS :=
 
 ifneq ($(NCCL_HOME), "")
-HIPCUFLAGS += -I$(NCCL_HOME)
+HIPCUFLAGS += -I$(NCCL_HOME) -I$(NCCL_HOME)/rccl/include
 HIPLDFLAGS   += -Wl,-rpath,$(NCCL_HOME) -L$(NCCL_HOME)
 endif
 HIPCUFLAGS += -I$(ROCM_PATH)/include