Merge pull request #31 from wzamazon/fix_makefile

Add -L$(MPI_HOME)/lib64 to NVLDFLAGS
This commit is contained in:
Sylvain Jeaugey 2020-01-06 10:38:40 -08:00 committed by GitHub
commit c864b73a27
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -52,7 +52,7 @@ endif
ifeq ($(MPI), 1)
NVCUFLAGS += -DMPI_SUPPORT -I$(MPI_HOME)/include
NVLDFLAGS += -L$(MPI_HOME)/lib -lmpi
NVLDFLAGS += -L$(MPI_HOME)/lib -L$(MPI_HOME)/lib64 -lmpi
endif
LIBRARIES += curand nccl nvToolsExt
NVLDFLAGS += $(LIBRARIES:%=-l%)