add riscv support

Signed-off-by: craft <convercraft@gmail.com>

Author: craft-zhang

Makefile-rv

@@ -0,0 +1,128 @@
+# Copyright (C) 2020 Texas Instruments Incorporated - http://www.ti.com/
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+TARGET_TOOLCHAIN_PREFIX ?= aarch64-linux-gnu-
+
+TF_SRC_DIR              ?=/home/craft/workspace/gem5/tensorflow_src
+TFLITE_INC_DIR          ?=$(TF_SRC_DIR)
+TFLITE_BUILD_DIR        ?=$(TF_SRC_DIR)/build
+FLATBUFFERS_INC_DIR     ?=$(TFLITE_BUILD_DIR)/flatbuffers/include
+TFLITE_LIB_DIR          ?=$(TFLITE_BUILD_DIR)
+TFLITE_FLATBUF_LIB_DIR  ?=$(TFLITE_BUILD_DIR)/_deps/flatbuffers-build
+TFLITE_RUY_LIB_DIR      ?=$(TFLITE_BUILD_DIR)/_deps/ruy-build/ruy
+TFLITE_FARMHASH_LIB_DIR ?=$(TFLITE_BUILD_DIR)/_deps/farmhash-build
+TFLITE_FFT2D_DIR        ?=$(TFLITE_BUILD_DIR)/_deps/fft2d-build
+TFLITE_CPUINFO_DIR      ?=$(TFLITE_BUILD_DIR)/_deps/cpuinfo-build
+TFLITE_CLOG_DIR         ?=$(TFLITE_BUILD_DIR)/_deps/clog-build
+
+OPENCV_SRC_DIR          ?=/home/craft/workspace/gem5/opencv
+OPENCV_BUILD_DIR        ?=$(OPENCV_SRC_DIR)/build
+OPENCV_INC_DIR          ?=$(OPENCV_BUILD_DIR)/install/include/opencv4/
+OPENCV_LIB_DIR          ?=$(OPENCV_BUILD_DIR)/install/lib/
+OPENCV_LIB_3RDPARTY_DIR ?=$(OPENCV_BUILD_DIR)/install/lib/opencv4/3rdparty/
+
+PHONY: all
+
+all: classification/tflite_classification segmentation/tflite_segmentation detection/tflite_detection
+
+LIBS  = $(TFLITE_LIB_DIR)/libtensorflow-lite.a
+LIBS += $(TFLITE_RUY_LIB_DIR)/libruy_ctx.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_frontend.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_tune.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_context.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_context_get_ctx.a \
+	\
+	$(TFLITE_RUY_LIB_DIR)/libruy_thread_pool.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_wait.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_allocator.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_system_aligned_alloc.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_cpuinfo.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_blocking_counter.a \
+	\
+	$(TFLITE_RUY_LIB_DIR)/libruy_apply_multiplier.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_trmul.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_kernel_arm.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_block_map.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_pack_arm.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_prepacked_cache.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_prepare_packed_matrices.a \
+	$(TFLITE_RUY_LIB_DIR)/libruy_denormal.a
+
+LIBS += $(TFLITE_FLATBUF_LIB_DIR)/libflatbuffers.a
+LIBS += $(TFLITE_FARMHASH_LIB_DIR)/libfarmhash.a
+LIBS += $(TFLITE_FFT2D_DIR)/libfft2d_fftsg.a
+LIBS += $(TFLITE_FFT2D_DIR)/libfft2d_fftsg2d.a
+LIBS += $(TFLITE_CPUINFO_DIR)/libcpuinfo.a
+LIBS += $(TFLITE_CLOG_DIR)/libclog.a
+
+LIBS += $(OPENCV_LIB_DIR)/libopencv_features2d.a \
+	$(OPENCV_LIB_DIR)/libopencv_imgcodecs.a \
+	$(OPENCV_LIB_DIR)/libopencv_photo.a \
+	$(OPENCV_LIB_DIR)/libopencv_calib3d.a \
+	$(OPENCV_LIB_DIR)/libopencv_flann.a \
+	$(OPENCV_LIB_DIR)/libopencv_imgproc.a \
+	$(OPENCV_LIB_DIR)/libopencv_stitching.a \
+	$(OPENCV_LIB_DIR)/libopencv_gapi.a \
+	$(OPENCV_LIB_DIR)/libopencv_ml.a \
+	$(OPENCV_LIB_DIR)/libopencv_video.a \
+	$(OPENCV_LIB_DIR)/libopencv_dnn.a \
+	$(OPENCV_LIB_DIR)/libopencv_highgui.a \
+	$(OPENCV_LIB_DIR)/libopencv_objdetect.a \
+	$(OPENCV_LIB_DIR)/libopencv_videoio.a \
+	$(OPENCV_LIB_DIR)/libopencv_core.a
+
+LIBS += $(OPENCV_LIB_3RDPARTY_DIR)/liblibjpeg-turbo.a \
+	$(OPENCV_LIB_3RDPARTY_DIR)/liblibpng.a \
+	$(OPENCV_LIB_3RDPARTY_DIR)/libzlib.a \
+	$(OPENCV_LIB_3RDPARTY_DIR)/libade.a \
+	$(OPENCV_LIB_3RDPARTY_DIR)/liblibopenjp2.a \
+	$(OPENCV_LIB_3RDPARTY_DIR)/liblibprotobuf.a \
+	$(OPENCV_LIB_3RDPARTY_DIR)/liblibwebp.a \
+	$(OPENCV_LIB_3RDPARTY_DIR)/liblibtiff.a \
+	$(OPENCV_LIB_3RDPARTY_DIR)/libquirc.a
+
+LIBS += -lstdc++ -lpthread -lm -ldl -lrt -latomic -static
+
+INCLUDES := -I. -I$(FLATBUFFERS_INC_DIR) -I$(TFLITE_INC_DIR) -I$(OPENCV_INC_DIR)
+
+CCFLAGS = -O3 -DNDEBUG -fPIC
+
+CXX := ${TARGET_TOOLCHAIN_PREFIX}g++
+
+COMMON_SRC = model_utils.cc utils.cc
+
+JDSK_DL_ACC ?= no
+ifeq ($(JDSK_DL_ACC), yes)
+  COMMON_SRC +=
+  CCFLAGS += -march=rv64gcv0p7_zfh_xtheadc -mabi=lp64d -mtune=c906 -std=gnu++11
+
+  JDSK_API_LIB_DIR ?= /home/craft/workspace/DNN/XNNPACK-riscv/build/linux/c906-v240/install/lib/
+  LIBS += $(JDSK_API_LIB_DIR)/libXNNPACK.a \
+	$(JDSK_API_LIB_DIR)/libpthreadpool.a
+endif
+
+classification/tflite_classification: classification.cc $(COMMON_SRC)
+	mkdir -p classification
+	$(CXX) classification.cc $(COMMON_SRC) -o classification/tflite_classification $(LDFLAGS) $(LIBS) $(CXXFLAGS) $(CCFLAGS) $(INCLUDES)
+
+segmentation/tflite_segmentation: segmentation.cc $(COMMON_SRC)
+	mkdir -p segmentation
+	$(CXX) segmentation.cc $(COMMON_SRC) -o segmentation/tflite_segmentation $(LDFLAGS) $(LIBS) $(CXXFLAGS) $(CCFLAGS) $(INCLUDES)
+
+detection/tflite_detection: detection.cc yolov5.cc $(COMMON_SRC)
+	mkdir -p detection
+	$(CXX) detection.cc yolov5.cc $(COMMON_SRC) -o detection/tflite_detection $(LDFLAGS) $(LIBS) $(CXXFLAGS) $(CCFLAGS) $(INCLUDES)
+
+clean:
+	rm -rf classification/tflite_classification segmentation/tflite_segmentation detection/tflite_detection

scripts/make_riscv.sh

@@ -0,0 +1,17 @@
+#!/usr/bin/env bash
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+set -e
+
+export TARGET_TOOLCHAIN_PREFIX=riscv64-unknown-linux-gnu-
+
+export TFLITE_BUILD_DIR=/home/craft/workspace/gem5/tensorflow_src/buildrv64
+
+export OPENCV_SRC_DIR=/home/craft/workspace/gem5/opencv
+export OPENCV_BUILD_DIR=/home/craft/workspace/gem5/opencv/buildrv64
+
+export JDSK_DL_ACC=yes
+
+make