tflite compile script

Signed-off-by: zhangdanfeng <craft.zhang@space-t.cn>

Author: zhangdanfeng

scripts/make_riscv.sh

@@ -5,6 +5,8 @@
 
 set -e
 
+make clean
+
 export TARGET_TOOLCHAIN_PREFIX=riscv64-unknown-linux-gnu-
 
 export TFLITE_BUILD_DIR=/home/craft/workspace/gem5/tensorflow_src/buildrv64

scripts/tf/build-linux-riscv64-JDSK.sh

@@ -0,0 +1,53 @@
+#!/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.
+# tf v2.9.1 tested
+
+set -e
+
+# change to your path
+mkdir -p build_tflite
+
+CMAKE_ARGS=()
+
+# CMake-level configuration
+CMAKE_ARGS+=("-DCMAKE_MAKE_PROGRAM=make")
+CMAKE_ARGS+=("-DCMAKE_CROSSCOMPILING=1")
+CMAKE_ARGS+=("-DCMAKE_TOOLCHAIN_FILE=$(pwd)/scripts/tf/cmake/riscv64.JDSK.toolchain.cmake")
+CMAKE_ARGS+=("-DCMAKE_BUILD_TYPE=RelWithDebInfo")
+
+# Cross-compilation options for Google Benchmark
+
+CMAKE_ARGS+=("-DHAVE_POSIX_REGEX=0")
+CMAKE_ARGS+=("-DHAVE_STEADY_CLOCK=0")
+CMAKE_ARGS+=("-DHAVE_STD_REGEX=0")
+
+# install
+CMAKE_ARGS+=("-DCMAKE_INSTALL_PREFIX=$(pwd)/tflite")
+CMAKE_ARGS+=("-DTFLITE_ENABLE_INSTALL=ON")
+ABSL_ENABLE_INSTALL
+CMAKE_ARGS+=("-DCMAKE_FIND_PACKAGE_PREFER_CONFIG=ON")
+CMAKE_ARGS+=("-Dabsl_DIR=$(pwd)/tflite/lib/cmake/absl")
+CMAKE_ARGS+=("-DEigen3_DIR=$(pwd)/tflite/share/eigen3/cmake")
+CMAKE_ARGS+=("-DFlatbuffers_DIR=$(pwd)/tflite/lib/cmake/flatbuffers")
+CMAKE_ARGS+=("-DNEON_2_SSE_DIR=$(pwd)/tflite/lib/cmake/NEON_2_SSE")
+CMAKE_ARGS+=("-Dcpuinfo_DIR=$(pwd)/tflite/share/cpuinfo")
+CMAKE_ARGS+=("-Druy_DIR=$(pwd)/tflite/lib/cmake/ruy")
+
+# xnnpack
+CMAKE_ARGS+=("-DTFLITE_ENABLE_XNNPACK=ON")
+
+# Use-specified CMake arguments go last to allow overridding defaults
+CMAKE_ARGS+=($@)
+
+# change to your path
+pushd build_tflite
+
+ccmake ../../tensorflow/tensorflow/lite "${CMAKE_ARGS[@]}"
+
+cmake --build .
+
+popd
+
+# cp 

scripts/tf/cmake/aarch64-linux-gnu.toolchain.cmake

@@ -0,0 +1,22 @@
+#
+# CMake Toolchain file for crosscompiling on ARM.
+#
+# Target operating system name.
+set(CMAKE_SYSTEM_NAME Linux)
+set(CMAKE_SYSTEM_PROCESSOR aarch64)
+set(CMAKE_CROSSCOMPILING TRUE)
+
+set(CMAKE_C_COMPILER "aarch64-linux-gnu-gcc")
+set(CMAKE_CXX_COMPILER "aarch64-linux-gnu-g++")
+
+set(CMAKE_FIND_ROOT_PATH_MODE_PROGRAM NEVER)
+set(CMAKE_FIND_ROOT_PATH_MODE_LIBRARY ONLY)
+set(CMAKE_FIND_ROOT_PATH_MODE_INCLUDE ONLY)
+
+# 增加SVE支持
+set(CMAKE_C_FLAGS "-march=armv8-a+sve")
+set(CMAKE_CXX_FLAGS "-march=armv8-a+sve")
+
+# cache flags
+set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS}" CACHE STRING "c flags")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}" CACHE STRING "c++ flags")

scripts/tf/cmake/riscv64.JDSK.toolchain.cmake

@@ -0,0 +1,48 @@
+#
+# CMake Toolchain file for crosscompiling on RV64.
+#
+# Target operating system name.
+#
+# usage: cmake -DCMAKE_TOOLCHAIN_FILE=riscv64.JDSK.toolchain.cmake
+# 
+# static link
+#
+
+set(CMAKE_SYSTEM_NAME Linux)
+set(CMAKE_SYSTEM_PROCESSOR riscv64)
+
+if(DEFINED ENV{RISCV_ROOT_PATH})
+    file(TO_CMAKE_PATH $ENV{RISCV_ROOT_PATH} RISCV_ROOT_PATH)
+else()
+    message(FATAL_ERROR "RISCV_ROOT_PATH env must be defined")
+endif()
+
+set(RISCV_ROOT_PATH ${RISCV_ROOT_PATH} CACHE STRING "root path to riscv toolchain")
+
+set(CMAKE_C_COMPILER "${RISCV_ROOT_PATH}/bin/riscv64-unknown-linux-gnu-gcc")
+set(CMAKE_CXX_COMPILER "${RISCV_ROOT_PATH}/bin/riscv64-unknown-linux-gnu-g++")
+
+set(CMAKE_FIND_ROOT_PATH "${RISCV_ROOT_PATH}/riscv64-unknown-linux-gnu")
+set(CMAKE_SYSROOT "${RISCV_ROOT_PATH}/sysroot")
+
+set(CMAKE_FIND_ROOT_PATH_MODE_PROGRAM NEVER)
+set(CMAKE_FIND_ROOT_PATH_MODE_LIBRARY ONLY)
+set(CMAKE_FIND_ROOT_PATH_MODE_INCLUDE ONLY)
+set(CMAKE_FIND_ROOT_PATH_MODE_PACKAGE ONLY)
+
+# march
+set(CMAKE_C_FLAGS "-march=rv64gcv -mabi=lp64d")
+set(CMAKE_CXX_FLAGS "-march=rv64gcv -mabi=lp64d")
+
+# c/c++ std
+set(CMAKE_CXX_FLAGS "-std=gnu++11 ${CMAKE_CXX_FLAGS}")
+set(CMAKE_C_FLAGS "-std=c99 ${CMAKE_C_FLAGS}")
+
+# cache flags
+set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS}" CACHE STRING "c flags")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}" CACHE STRING "c++ flags")
+
+set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -static")
+
+FIND_LIBRARY(LIBM libm.a)
+FIND_LIBRARY(LIBRT librt.a)