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

add the rccl/lib directory to the link path
Cette révision appartient à :
Edgar Gabriel
2022-10-31 15:54:55 -05:00
révisé par GitHub
révision a80fbba12b
+1 -1
Voir le fichier
@@ -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