-
Notifications
You must be signed in to change notification settings - Fork 0
/
makefile
96 lines (68 loc) · 2.94 KB
/
makefile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
.SUFFIXES = .cpp .o
#CFLAGS = -O2 -std=c++11
CFLAGS = -std=c++11 -g
WICWIU_LIB = lib/libwicwiu.a
# if CUDA device, cuda or cuDNN is not installed, disable the following line
ENABLE_CUDNN = -D__CUDNN__
# uncomment the following to debug
#DFLAGS = -D__DEBUG__
INCLUDE_PATH = -I/usr/local/cuda/include
LIB_PATH = -L. -L/usr/local/cuda/lib64
CC = g++
NVCC = nvcc
ifdef ENABLE_CUDNN
LINKER = nvcc
LFLAGS = -lcudart -lcudnn -lpthread
# LFLAGS = -lpthread
else
LINKER = g++
LFLAGS = -lpthread
endif
AR = ar
WICWIU_SRCS = \
WICWIU_src/Utils.cpp \
WICWIU_src/Shape.cpp \
WICWIU_src/KNearestNeighbor.cpp \
WICWIU_src/FewShotClassifier.cpp
WICWIU_OBJS = ${WICWIU_SRCS:.cpp=.o}
ifdef ENABLE_CUDNN
WICWIU_CUDA_SRCS = \
WICWIU_src/Utils_CUDA.cu \
WICWIU_src/Optimizer/AdamOptimizer_CUDA.cu \
WICWIU_src/Operator/Concatenate_CUDA.cu \
WICWIU_src/Operator/LRelu_CUDA.cu \
WICWIU_src/Operator/PRelu_CUDA.cu\
WICWIU_src/Optimizer/NagOptimizer_CUDA.cu \
WICWIU_src/Optimizer/AdagradOptimizer_CUDA.cu \
WICWIU_src/Optimizer/RMSPropOptimizer_CUDA.cu \
WICWIU_src/LossFunction/SoftmaxCrossEntropy_CUDA.cu
WICWIU_CUDA_OBJS = ${WICWIU_CUDA_SRCS:.cu=.o}
endif
all: $(WICWIU_LIB)
.cpp.o:
$(CC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) $(LIB_PATH) -c $< -o $@
# for cuda code
WICWIU_src/Utils_CUDA.o: WICWIU_src/Utils_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
WICWIU_src/Optimizer/AdamOptimizer_CUDA.o: WICWIU_src/Optimizer/AdamOptimizer_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
WICWIU_src/Optimizer/RMSPropOptimizer_CUDA.o: WICWIU_src/Optimizer/RMSPropOptimizer_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
WICWIU_src/Optimizer/NagOptimizer_CUDA.o: WICWIU_src/Optimizer/NagOptimizer_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
WICWIU_src/Optimizer/AdagradOptimizer_CUDA.o: WICWIU_src/Optimizer/AdagradOptimizer_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
WICWIU_src/Operator/Concatenate_CUDA.o: WICWIU_src/Operator/Concatenate_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
WICWIU_src/Operator/LRelu_CUDA.o: WICWIU_src/Operator/LRelu_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
WICWIU_src/Operator/PRelu_CUDA.o: WICWIU_src/Operator/PRelu_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
WICWIU_src/LossFunction/SoftmaxCrossEntropy_CUDA.o: WICWIU_src/LossFunction/SoftmaxCrossEntropy_CUDA.cu
$(NVCC) $(CFLAGS) $(DFLAGS) $(ENABLE_CUDNN) $(INCLUDE_PATH) -c $< -o $@
$(WICWIU_LIB): $(WICWIU_OBJS) $(WICWIU_CUDA_OBJS)
$(AR) rcs $@ $(WICWIU_OBJS) $(WICWIU_CUDA_OBJS)
#main: $(WICWIU_OBJS) main.o
# $(LINKER) $(CFLAGS) $(ENABLE_CUDNN) $(DFLAGS) $(LFLAGS) $(INCLUDE_PATH) $(LIB_PATH) -o $@ $(WICWIU_OBJS) main.o
clean:
rm -rf *.o $(WICWIU_OBJS) $(WICWIU_CUDA_OBJS) $(WICWIU_LIB)