ncnn
41 строка · 1.6 Кб
1set(CMAKE_SYSTEM_NAME Linux)
2set(CMAKE_SYSTEM_PROCESSOR riscv64)
3set(C906 True)
4
5if(DEFINED ENV{RISCV_ROOT_PATH})
6file(TO_CMAKE_PATH $ENV{RISCV_ROOT_PATH} RISCV_ROOT_PATH)
7else()
8message(FATAL_ERROR "RISCV_ROOT_PATH env must be defined")
9endif()
10
11set(RISCV_ROOT_PATH ${RISCV_ROOT_PATH} CACHE STRING "root path to riscv toolchain")
12
13set(CMAKE_C_COMPILER "${RISCV_ROOT_PATH}/bin/riscv64-unknown-linux-gnu-gcc")
14set(CMAKE_CXX_COMPILER "${RISCV_ROOT_PATH}/bin/riscv64-unknown-linux-gnu-g++")
15
16set(CMAKE_FIND_ROOT_PATH "${RISCV_ROOT_PATH}/riscv64-unknown-linux-gnu")
17
18set(CMAKE_SYSROOT "${RISCV_ROOT_PATH}/sysroot")
19
20if(NOT CMAKE_FIND_ROOT_PATH_MODE_PROGRAM)
21set(CMAKE_FIND_ROOT_PATH_MODE_PROGRAM NEVER)
22endif()
23if(NOT CMAKE_FIND_ROOT_PATH_MODE_LIBRARY)
24set(CMAKE_FIND_ROOT_PATH_MODE_LIBRARY ONLY)
25endif()
26if(NOT CMAKE_FIND_ROOT_PATH_MODE_INCLUDE)
27set(CMAKE_FIND_ROOT_PATH_MODE_INCLUDE ONLY)
28endif()
29if(NOT CMAKE_FIND_ROOT_PATH_MODE_PACKAGE)
30set(CMAKE_FIND_ROOT_PATH_MODE_PACKAGE ONLY)
31endif()
32
33set(CMAKE_C_FLAGS "-march=rv64gcv0p7_zfh_xtheadc -mabi=lp64d -mtune=c906 -DC906=1 -static")
34set(CMAKE_CXX_FLAGS "-march=rv64gcv0p7_zfh_xtheadc -mabi=lp64d -mtune=c906 -DC906=1 -static")
35
36# cache flags
37set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS}" CACHE STRING "c flags")
38set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}" CACHE STRING "c++ flags")
39
40# export RISCV_ROOT_PATH=/home/nihui/osd/Xuantie-900-gcc-linux-5.10.4-glibc-x86_64-V2.2.2
41# cmake -DCMAKE_TOOLCHAIN_FILE=../toolchains/c906-v222.toolchain.cmake -DCMAKE_BUILD_TYPE=release -DNCNN_BUILD_TESTS=ON -DNCNN_OPENMP=OFF -DNCNN_THREADS=OFF -DNCNN_RUNTIME_CPU=OFF -DNCNN_RVV=ON -DNCNN_SIMPLEOCV=ON -DNCNN_BUILD_EXAMPLES=ON ..
42