Update include path for custom RCCL build

[ROCm/rccl-tests commit: 47238336d9]
This commit is contained in:
Wenkai Du
2022-03-31 13:18:02 -04:00
parent 08bee52b97
commit dded9afea8
+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