forked from rioyokotalab/nbd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Makefile
23 lines (18 loc) · 873 Bytes
/
Makefile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
CXX = mpicxx -std=c++14
CFLAGS = -O3 -m64 -Wall -Wextra -I. -I"${MKLROOT}/include" -fopenmp
LDFLAGS = -L${MKLROOT}/lib/intel64 -lmkl_intel_lp64 -lmkl_sequential -lmkl_core -lpthread -lm -ldl
NVCC = ${CUDADIR}/bin/nvcc -ccbin=mpicxx -std=c++14
NVCCFLAGS = -O3 -m64 -I. -I"${MKLROOT}/include" --Werror=all-warnings -Xcompiler=-fopenmp
NVCCLIBS = -L${CUDADIR}/lib64 -lcublas -lcudart -lcusolver -lnccl
all:
$(CXX) $(CFLAGS) -c linalg.cxx -o linalg.o
$(CXX) $(CFLAGS) -c basis.cxx -o basis.o
$(CXX) $(CFLAGS) -c build_tree.cxx -o build_tree.o
$(CXX) $(CFLAGS) -c comm.cxx -o comm.o
$(CXX) $(CFLAGS) -c umv.cxx -o umv.o
$(CXX) $(CFLAGS) -c lorasp.cxx -o lorasp.o
$(NVCC) $(NVCCFLAGS) -c kernel.cu -o kernel.o
$(NVCC) $(NVCCFLAGS) linalg.o basis.o build_tree.o comm.o umv.o kernel.o \
lorasp.o $(LDFLAGS) $(NVCCLIBS) -o lorasp
clean:
rm -rf *.a *.o lorasp