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

add the rccl/lib directory to the link path
Tento commit je obsažen v:
Edgar Gabriel
2022-10-31 15:54:55 -05:00
odevzdal GitHub
+1 -1
Zobrazit soubor
@@ -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