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

View File

@ -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)