Update include path for custom RCCL build

This commit is contained in:
Wenkai Du
2022-03-31 13:18:02 -04:00
parent 698524e42e
commit 47238336d9
+1 -1
View File
@@ -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