Fix NCCL_HOME to be consistent with README

This commit is contained in:
Sylvain Jeaugey
2017-08-09 10:41:31 -07:00
parent 9ec3e35276
commit 25016c8eeb
+3 -3
View File
@@ -43,9 +43,9 @@ endif
.PHONY: build clean .PHONY: build clean
BUILDDIR ?= ../build BUILDDIR ?= ../build
ifneq ($(NCCLDIR), "") ifneq ($(NCCL_HOME), "")
NVCUFLAGS += -I$(NCCLDIR)/include/ NVCUFLAGS += -I$(NCCL_HOME)/include/
NVLDFLAGS += -L$(NCCLDIR)/lib NVLDFLAGS += -L$(NCCL_HOME)/lib
endif endif
ifeq ($(MPI), 1) ifeq ($(MPI), 1)