bda/gpu-mpi-tests
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use define to switch between managed/unmanaged

Browse Source
This commit is contained in:
Bryce Allen
2020-08-07 14:14:56 -04:00
parent 6940ce7ceb
commit 37ad5e87ce
2 changed files with 84 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Makefile
View File

@@ -1,5 +1,5 @@
.PHONY: all
all: daxpy mpi_daxpy mpienv daxpy_nvtx mpi_daxpy_nvtx
all: daxpy mpi_daxpy mpienv daxpy_nvtx mpi_daxpy_nvtx_managed mpi_daxpy_nvtx_unmanaged
CCFLAGS = -std=c++11
CUDA_HOME ?= $(CUDA_DIR)
@@ -13,15 +13,18 @@ daxpy_nvtx: daxpy_nvtx.cu cuda_error.h
mpi_daxpy: mpi_daxpy.cc cuda_error.h
mpic++ $(CCFLAGS) -lcudart -lcublas -I$(CUDA_HOME)/include -L$(CUDA_HOME)/lib64 -o mpi_daxpy mpi_daxpy.cc
mpi_daxpy_nvtx: mpi_daxpy_nvtx.cc cuda_error.h
mpic++ $(CCFLAGS) -lcudart -lcublas -lnvToolsExt -I$(CUDA_HOME)/include -L$(CUDA_HOME)/lib64 -o mpi_daxpy_nvtx mpi_daxpy_nvtx.cc
mpi_daxpy_nvtx_managed: mpi_daxpy_nvtx.cc cuda_error.h
mpic++ $(CCFLAGS) -lcudart -lcublas -lnvToolsExt -I$(CUDA_HOME)/include -L$(CUDA_HOME)/lib64 -o mpi_daxpy_nvtx_managed mpi_daxpy_nvtx.cc -DMANAGED
mpi_daxpy_nvtx_unmanaged: mpi_daxpy_nvtx.cc cuda_error.h
mpic++ $(CCFLAGS) -lcudart -lcublas -lnvToolsExt -I$(CUDA_HOME)/include -L$(CUDA_HOME)/lib64 -o mpi_daxpy_nvtx_unmanaged mpi_daxpy_nvtx.cc
mpienv: mpienv.f90
mpif90 -o mpienv mpienv.f90
.PHONY: clean
clean:
rm -rf daxpy mpi_daxpy daxpy_nvtx mpi_daxpy_nvtx
rm -rf daxpy mpi_daxpy daxpy_nvtx mpi_daxpy_nvtx_managed mpi_daxpy_nvtx_unmanaged
.PHONY: force
force: clean all