-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmake-conv
14 lines (9 loc) · 989 Bytes
/
make-conv
1
2
3
4
5
6
7
8
9
10
11
12
13
14
CPP=g++
CFLAGS=$(OPT) --std=c++11 -O3
MODULE := conv1 conv2 class1 class2
.PHONY: all clean
all: $(MODULE)
HEADERS=dnn.hpp
# These tiling parameters are 100% arbitrary, and it may be advantageous to tune/remove/completely-change them for GPU
opt-conv1: convolution.cu
/usr/local/cuda-10.1/bin/nvcc -ccbin g++ -I /home/${USER}/NVIDIA_CUDA-10.1_Samples/common/inc -m64 -gencode arch=compute_30,code=sm_30 -gencode arch=compute_35,code=sm_35 -gencode arch=compute_37,code=sm_37 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_75,code=compute_75 $^ $(CFLAGS) -o $@ -DNx=${Nx} -DNy=${Ny} -DKx=3 -DKy=3 -DNi=${Ni} -DNn=${Nn} -DNUM_THREADS_Y=${NUM_THREADS_Y} -DNUM_BLOCKS_Y=${NUM_BLOCKS_Y} -DNUM_THREADS_X=${NUM_THREADS_X} -DNUM_BLOCKS_X=${NUM_BLOCKS_X} -DNUM_THREADS_Z=${NUM_THREADS_Z}