Fix NCCL_HOME to be consistent with README
This commit is contained in:
1 parent
9ec3e35276
commit
25016c8eeb
1 file changed
+3
-3
+3
-3
@@ -43,9 +43,9 @@ endif
|
||||
.PHONY: build clean
|
||||
|
||||
BUILDDIR ?= ../build
|
||||
ifneq ($(NCCLDIR), "")
|
||||
NVCUFLAGS += -I$(NCCLDIR)/include/
|
||||
NVLDFLAGS += -L$(NCCLDIR)/lib
|
||||
ifneq ($(NCCL_HOME), "")
|
||||
NVCUFLAGS += -I$(NCCL_HOME)/include/
|
||||
NVLDFLAGS += -L$(NCCL_HOME)/lib
|
||||
endif
|
||||
|
||||
ifeq ($(MPI), 1)
|
||||
|
||||
Reference in new issue
Block a user