add the rccl/lib directory to the link path

Цей коміт міститься в:
Edgar Gabriel
2022-10-31 19:01:22 +00:00
джерело fb0d339c1b
коміт 9c9746739a
+1 -1
Переглянути файл
@@ -21,7 +21,7 @@ HIPLDFLAGS :=
ifneq ($(NCCL_HOME), "")
HIPCUFLAGS += -I$(NCCL_HOME)/ -I$(NCCL_HOME)/include
HIPLDFLAGS += -Wl,-rpath,$(NCCL_HOME) -L$(NCCL_HOME)
HIPLDFLAGS += -Wl,-rpath,$(NCCL_HOME) -L$(NCCL_HOME) -L$(NCCL_HOME)/lib
endif
HIPCUFLAGS += -I$(ROCM_PATH)/include
HIPCUFLAGS += -I$(ROCM_PATH)/include/hip