Skip to content

Commit 6b37c49

Browse files
author
zhangdanfeng
committed
riscv compile
Signed-off-by: zhangdanfeng <[email protected]>
1 parent bd18745 commit 6b37c49

File tree

1 file changed

+10
-19
lines changed

1 file changed

+10
-19
lines changed

Makefile-rv

Lines changed: 10 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
TARGET_TOOLCHAIN_PREFIX ?= aarch64-linux-gnu-
15+
TARGET_TOOLCHAIN_PREFIX ?= riscv64-unknown-linux-gnu-
1616

17-
TF_SRC_DIR ?=/home/craft/workspace/gem5/tensorflow_src
17+
TF_SRC_DIR ?=/home/craft/workspace/DNN/tensorflow
1818
TFLITE_INC_DIR ?=$(TF_SRC_DIR)
19-
TFLITE_BUILD_DIR ?=$(TF_SRC_DIR)/build
19+
TFLITE_BUILD_DIR ?=build_tflite
2020
FLATBUFFERS_INC_DIR ?=$(TFLITE_BUILD_DIR)/flatbuffers/include
2121
TFLITE_LIB_DIR ?=$(TFLITE_BUILD_DIR)
2222
TFLITE_FLATBUF_LIB_DIR ?=$(TFLITE_BUILD_DIR)/_deps/flatbuffers-build
@@ -26,8 +26,11 @@ TFLITE_FFT2D_DIR ?=$(TFLITE_BUILD_DIR)/_deps/fft2d-build
2626
TFLITE_CPUINFO_DIR ?=$(TFLITE_BUILD_DIR)/_deps/cpuinfo-build
2727
TFLITE_CLOG_DIR ?=$(TFLITE_BUILD_DIR)/_deps/clog-build
2828

29+
TFLITE_XNNPACK_DIR ?=$(TFLITE_BUILD_DIR)/_deps/xnnpack-build
30+
TFLITE_PTHREADPOOL_DIR ?=$(TFLITE_BUILD_DIR)/pthreadpool
31+
2932
OPENCV_SRC_DIR ?=/home/craft/workspace/gem5/opencv
30-
OPENCV_BUILD_DIR ?=$(OPENCV_SRC_DIR)/build
33+
OPENCV_BUILD_DIR ?=build_cv
3134
OPENCV_INC_DIR ?=$(OPENCV_BUILD_DIR)/install/include/opencv4/
3235
OPENCV_LIB_DIR ?=$(OPENCV_BUILD_DIR)/install/lib/
3336
OPENCV_LIB_3RDPARTY_DIR ?=$(OPENCV_BUILD_DIR)/install/lib/opencv4/3rdparty/
@@ -59,6 +62,8 @@ LIBS += $(TFLITE_RUY_LIB_DIR)/libruy_ctx.a \
5962
$(TFLITE_RUY_LIB_DIR)/libruy_prepare_packed_matrices.a \
6063
$(TFLITE_RUY_LIB_DIR)/libruy_denormal.a
6164

65+
LIBS += /home/craft/workspace/DNN/XNNPACK/build/linux/JDSK/libXNNPACK.a
66+
LIBS += $(TFLITE_PTHREADPOOL_DIR)/libpthreadpool.a
6267
LIBS += $(TFLITE_FLATBUF_LIB_DIR)/libflatbuffers.a
6368
LIBS += $(TFLITE_FARMHASH_LIB_DIR)/libfarmhash.a
6469
LIBS += $(TFLITE_FFT2D_DIR)/libfft2d_fftsg.a
@@ -69,16 +74,12 @@ LIBS += $(TFLITE_CLOG_DIR)/libclog.a
6974
LIBS += $(OPENCV_LIB_DIR)/libopencv_features2d.a \
7075
$(OPENCV_LIB_DIR)/libopencv_imgcodecs.a \
7176
$(OPENCV_LIB_DIR)/libopencv_photo.a \
72-
$(OPENCV_LIB_DIR)/libopencv_calib3d.a \
7377
$(OPENCV_LIB_DIR)/libopencv_flann.a \
7478
$(OPENCV_LIB_DIR)/libopencv_imgproc.a \
75-
$(OPENCV_LIB_DIR)/libopencv_stitching.a \
76-
$(OPENCV_LIB_DIR)/libopencv_gapi.a \
7779
$(OPENCV_LIB_DIR)/libopencv_ml.a \
7880
$(OPENCV_LIB_DIR)/libopencv_video.a \
7981
$(OPENCV_LIB_DIR)/libopencv_dnn.a \
8082
$(OPENCV_LIB_DIR)/libopencv_highgui.a \
81-
$(OPENCV_LIB_DIR)/libopencv_objdetect.a \
8283
$(OPENCV_LIB_DIR)/libopencv_videoio.a \
8384
$(OPENCV_LIB_DIR)/libopencv_core.a
8485

@@ -96,22 +97,12 @@ LIBS += -lstdc++ -lpthread -lm -ldl -lrt -latomic -static
9697

9798
INCLUDES := -I. -I$(FLATBUFFERS_INC_DIR) -I$(TFLITE_INC_DIR) -I$(OPENCV_INC_DIR)
9899

99-
CCFLAGS = -O3 -DNDEBUG -fPIC
100+
CCFLAGS = -O3 -DNDEBUG -fPIC -march=rv64gcv -mabi=lp64d -std=gnu++11
100101

101102
CXX := ${TARGET_TOOLCHAIN_PREFIX}g++
102103

103104
COMMON_SRC = model_utils.cc utils.cc
104105

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-
115106
classification/tflite_classification: classification.cc $(COMMON_SRC)
116107
mkdir -p classification
117108
$(CXX) classification.cc $(COMMON_SRC) -o classification/tflite_classification $(LDFLAGS) $(LIBS) $(CXXFLAGS) $(CCFLAGS) $(INCLUDES)

0 commit comments

Comments
 (0)