Fix NCCL_HOME to be consistent with README

2017-08-09 10:41:31 -07:00
Commit 25016c8eeb
@@ -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)