Fix NCCL_HOME to be consistent with README

2017-08-09 10:41:31 -07:00
1 parent 9ec3e35276
commit 25016c8eeb
@@ -43,9 +43,9 @@ endif
 .PHONY: build clean
 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
 ifeq ($(MPI), 1)