add cublas to makefile

This commit is contained in:
Ruipeng Li 2022-03-10 08:43:44 -08:00
parent 8ecee0b47d
commit 7681f7f180

View File

@ -25,7 +25,7 @@ HYPRE_DIR = ../hypre
########################################################################
ifeq ($(use_cuda), 1)
CUDA_INCL = -I${CUDA_HOME}/include
CUDA_LIBS = -L${CUDA_HOME}/lib64 -lcudart -lcusparse -lcurand -lstdc++ -L$(XL_DIR)/xlC/16.1.1/lib -libmc++
CUDA_LIBS = -L${CUDA_HOME}/lib64 -lcudart -lcublas -lcusparse -lcurand -lstdc++ -L$(XL_DIR)/xlC/16.1.1/lib -libmc++
CUDA_ARCH = -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70
NVCC_LDFLAGS = -ccbin=${CXX} ${CUDA_ARCH}
COPTS_CUDA = -DHYPRE_EXAMPLE_USING_CUDA
@ -33,14 +33,14 @@ ifeq ($(use_cuda), 1)
endif
ifeq ($(use_cuf), 1)
CUDA_LIBS = -L${CUDA_HOME}/lib64 -lcudart -lcusparse -lcurand -lstdc++ -L$(XL_DIR)/xlC/16.1.1/lib -libmc++
CUDA_LIBS = -L${CUDA_HOME}/lib64 -lcudart -lcublas -lcusparse -lcurand -lstdc++ -L$(XL_DIR)/xlC/16.1.1/lib -libmc++
endif
########################################################################
# Device OMP
########################################################################
ifeq ($(use_domp), 1)
DOMP_LIBS = -L${CUDA_HOME}/lib64 -lcudart -lcusparse -lcurand -lstdc++ -L$(XL_DIR)/xlC/16.1.1/lib -libmc++
DOMP_LIBS = -L${CUDA_HOME}/lib64 -lcudart -lcublas -lcusparse -lcurand -lstdc++ -L$(XL_DIR)/xlC/16.1.1/lib -libmc++
COPTS_DOMP = -DHYPRE_EXAMPLE_USING_DEVICE_OMP
FOPTS_DOMP = -qoffload -W@,-v -qsmp=omp -qinfo=omperrtrace -DHYPRE_EXAMPLE_USING_DEVICE_OMP
LOPTS_DOMP = -qoffload -W@,-v -qsmp=omp