Merge pull request #23 from edgargabriel/pr/link-fix

add the rccl/lib directory to the link path
Этот коммит содержится в:
Edgar Gabriel
2022-10-31 15:54:55 -05:00
коммит произвёл GitHub
родитель fb0d339c1b 9c9746739a
Коммит a80fbba12b
+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