12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
15
- TARGET_TOOLCHAIN_PREFIX ?= aarch64 -linux-gnu-
15
+ TARGET_TOOLCHAIN_PREFIX ?= riscv64-unknown -linux-gnu-
16
16
17
- TF_SRC_DIR ?=/home/craft/workspace/gem5/tensorflow_src
17
+ TF_SRC_DIR ?=/home/craft/workspace/DNN/tensorflow
18
18
TFLITE_INC_DIR ?=$(TF_SRC_DIR)
19
- TFLITE_BUILD_DIR ?=$(TF_SRC_DIR)/build
19
+ TFLITE_BUILD_DIR ?=build_tflite
20
20
FLATBUFFERS_INC_DIR ?=$(TFLITE_BUILD_DIR)/flatbuffers/include
21
21
TFLITE_LIB_DIR ?=$(TFLITE_BUILD_DIR)
22
22
TFLITE_FLATBUF_LIB_DIR ?=$(TFLITE_BUILD_DIR)/_deps/flatbuffers-build
@@ -26,8 +26,11 @@ TFLITE_FFT2D_DIR ?=$(TFLITE_BUILD_DIR)/_deps/fft2d-build
26
26
TFLITE_CPUINFO_DIR ?=$(TFLITE_BUILD_DIR)/_deps/cpuinfo-build
27
27
TFLITE_CLOG_DIR ?=$(TFLITE_BUILD_DIR)/_deps/clog-build
28
28
29
+ TFLITE_XNNPACK_DIR ?=$(TFLITE_BUILD_DIR)/_deps/xnnpack-build
30
+ TFLITE_PTHREADPOOL_DIR ?=$(TFLITE_BUILD_DIR)/pthreadpool
31
+
29
32
OPENCV_SRC_DIR ?=/home/craft/workspace/gem5/opencv
30
- OPENCV_BUILD_DIR ?=$(OPENCV_SRC_DIR)/build
33
+ OPENCV_BUILD_DIR ?=build_cv
31
34
OPENCV_INC_DIR ?=$(OPENCV_BUILD_DIR)/install/include/opencv4/
32
35
OPENCV_LIB_DIR ?=$(OPENCV_BUILD_DIR)/install/lib/
33
36
OPENCV_LIB_3RDPARTY_DIR ?=$(OPENCV_BUILD_DIR)/install/lib/opencv4/3rdparty/
@@ -59,6 +62,8 @@ LIBS += $(TFLITE_RUY_LIB_DIR)/libruy_ctx.a \
59
62
$(TFLITE_RUY_LIB_DIR)/libruy_prepare_packed_matrices.a \
60
63
$(TFLITE_RUY_LIB_DIR)/libruy_denormal.a
61
64
65
+ LIBS += /home/craft/workspace/DNN/XNNPACK/build/linux/JDSK/libXNNPACK.a
66
+ LIBS += $(TFLITE_PTHREADPOOL_DIR)/libpthreadpool.a
62
67
LIBS += $(TFLITE_FLATBUF_LIB_DIR)/libflatbuffers.a
63
68
LIBS += $(TFLITE_FARMHASH_LIB_DIR)/libfarmhash.a
64
69
LIBS += $(TFLITE_FFT2D_DIR)/libfft2d_fftsg.a
@@ -69,16 +74,12 @@ LIBS += $(TFLITE_CLOG_DIR)/libclog.a
69
74
LIBS += $(OPENCV_LIB_DIR)/libopencv_features2d.a \
70
75
$(OPENCV_LIB_DIR)/libopencv_imgcodecs.a \
71
76
$(OPENCV_LIB_DIR)/libopencv_photo.a \
72
- $(OPENCV_LIB_DIR)/libopencv_calib3d.a \
73
77
$(OPENCV_LIB_DIR)/libopencv_flann.a \
74
78
$(OPENCV_LIB_DIR)/libopencv_imgproc.a \
75
- $(OPENCV_LIB_DIR)/libopencv_stitching.a \
76
- $(OPENCV_LIB_DIR)/libopencv_gapi.a \
77
79
$(OPENCV_LIB_DIR)/libopencv_ml.a \
78
80
$(OPENCV_LIB_DIR)/libopencv_video.a \
79
81
$(OPENCV_LIB_DIR)/libopencv_dnn.a \
80
82
$(OPENCV_LIB_DIR)/libopencv_highgui.a \
81
- $(OPENCV_LIB_DIR)/libopencv_objdetect.a \
82
83
$(OPENCV_LIB_DIR)/libopencv_videoio.a \
83
84
$(OPENCV_LIB_DIR)/libopencv_core.a
84
85
@@ -96,22 +97,12 @@ LIBS += -lstdc++ -lpthread -lm -ldl -lrt -latomic -static
96
97
97
98
INCLUDES := -I. -I$(FLATBUFFERS_INC_DIR) -I$(TFLITE_INC_DIR) -I$(OPENCV_INC_DIR)
98
99
99
- CCFLAGS = -O3 -DNDEBUG -fPIC
100
+ CCFLAGS = -O3 -DNDEBUG -fPIC -march=rv64gcv -mabi=lp64d -std=gnu++11
100
101
101
102
CXX := ${TARGET_TOOLCHAIN_PREFIX}g++
102
103
103
104
COMMON_SRC = model_utils.cc utils.cc
104
105
105
- JDSK_DL_ACC ?= no
106
- ifeq ($(JDSK_DL_ACC), yes)
107
- COMMON_SRC +=
108
- CCFLAGS += -march=rv64gcv0p7_zfh_xtheadc -mabi=lp64d -mtune=c906 -std=gnu++11
109
-
110
- JDSK_API_LIB_DIR ?= /home/craft/workspace/DNN/XNNPACK-riscv/build/linux/c906-v240/install/lib/
111
- LIBS += $(JDSK_API_LIB_DIR)/libXNNPACK.a \
112
- $(JDSK_API_LIB_DIR)/libpthreadpool.a
113
- endif
114
-
115
106
classification/tflite_classification: classification.cc $(COMMON_SRC)
116
107
mkdir -p classification
117
108
$(CXX) classification.cc $(COMMON_SRC) -o classification/tflite_classification $(LDFLAGS) $(LIBS) $(CXXFLAGS) $(CCFLAGS) $(INCLUDES)
0 commit comments