26 lines
606 B
Makefile
26 lines
606 B
Makefile
TARGET=convolution
|
|
OBJECTS=main.o util.o convolution_cpu.o convolution_gpu.o convolution_cudnn.o
|
|
|
|
CPPFLAGS=-std=c++14 -O3 -Wall -march=native -mavx2 -mfma -fopenmp -mno-avx512f -I/usr/local/cuda/include
|
|
CUDA_CFLAGS:=$(foreach option, $(CPPFLAGS),-Xcompiler=$(option))
|
|
|
|
LDFLAGS=-L/usr/local/cuda/lib64
|
|
LDLIBS=-lstdc++ -lcudart -lm -lcudnn
|
|
|
|
CXX=g++
|
|
CUX=/usr/local/cuda/bin/nvcc
|
|
|
|
all: $(TARGET)
|
|
|
|
$(TARGET): $(OBJECTS)
|
|
$(CC) $(CPPFLAGS) -o $(TARGET) $(OBJECTS) $(LDFLAGS) $(LDLIBS)
|
|
|
|
%.o: %.cpp
|
|
$(CXX) $(CPPFLAGS) -c -o $@ $^
|
|
|
|
%.o: %.cu
|
|
$(CUX) $(CUDA_CFLAGS) -c -o $@ $^
|
|
|
|
clean:
|
|
rm -rf $(TARGET) $(OBJECTS)
|