Skip to content

Commit

Permalink
NN2: version 1.12.10
Browse files Browse the repository at this point in the history
  • Loading branch information
@zhangwm-pt
zhangwm-pt committed Apr 14, 2022
1 parent 3317645 commit 0e784c3
Show file tree
Hide file tree
Showing 1,685 changed files with 61,212 additions and 30,686 deletions.
254 changes: 254 additions & 0 deletions CMakeLists.txt
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,254 @@
cmake_minimum_required(VERSION 3.3)
project(CSI-NN2)

enable_language(ASM)

option(USE_CSI_NN2_DEBUG "option for debug" ON)

option(BUILD_X86 "build x86" OFF)
option(BUILD_RISCV "build riscv" OFF)
option(BUILD_RISCV_ELF "build riscv elf" OFF)
option(BUILD_CSKY "build csky" OFF)
option(BUILD_CSKY_ELF "build csky elf" OFF)

# riscv linux compiler
if (BUILD_RISCV)
set(CMAKE_C_COMPILER riscv64-unknown-linux-gnu-gcc)
set(CMAKE_CXX_COMPILER riscv64-unknown-linux-gnu-g++)
endif()

# riscv elf compiler
if (BUILD_RISCV_ELF)
set(CMAKE_C_COMPILER riscv64-unknown-elf-gcc)
endif()

# csky linux compiler
if (BUILD_CSKY)
set(CMAKE_C_COMPILER csky-abiv2-linux-gcc)
set(CMAKE_ASM_COMPILER csky-abiv2-linux-gcc)
endif()

# csky elf compiler
if (BUILD_CSKY_ELF)
set(CMAKE_C_COMPILER csky-abiv2-elf-gcc)
set(CMAKE_ASM_COMPILER csky-abiv2-elf-gcc)
endif()

# CSI-NN2 debug module
if(USE_CSI_NN2_DEBUG)
add_definitions(-D CSI_DEBUG)
endif()

# reduce elf size
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -ffunction-sections -fdata-sections -Wl,--gc-sections")

# set warning as error
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -Werror")

file(GLOB_RECURSE NN2_SRCS source/nn2/*.c source/utils/*.c)
file(GLOB_RECURSE REF_SRCS source/reference/*.c)
file(GLOB_RECURSE GREF_SRCS source/graph_ref/*.c)
file(GLOB_RECURSE OPENVX_SRCS source/openvx/*.c)
file(GLOB_RECURSE PNNA_SRCS source/pnna/*.c source/pnna/*.cpp)
file(GLOB_RECURSE C906_SRCS source/c906_opt/*.c)
file(GLOB_RECURSE C908_SRCS source/c908/*.c)
file(GLOB_RECURSE THEAD_RVV_SRCS source/thead_rvv/*.c)
file(GLOB_RECURSE C860_SRCS source/c860_opt/*.S)
file(GLOB_RECURSE I805_REF_SRCS source/i805_ref/*.c)
file(GLOB_RECURSE I805_SRCS source/i805_opt/*.c source/i805_opt/*.S)
file(GLOB_RECURSE E804_SRCS source/e804_opt/*.c source/e804_opt/*.S)
file(GLOB_RECURSE CH8601_SRCS source/ch8601/*.c)
file(GLOB_RECURSE DP1K_SRCS source/dp1k/*.c)
file(GLOB_RECURSE ASP_SRCS source/asp/*.c)

include_directories(include)

option(CSINN_LAYER_BENCHMARK "Layer information and performance" OFF)
if(CSINN_LAYER_BENCHMARK)
add_definitions(-DCSINN_LAYER_BENCHMARK)
message(STATUS "Print the execution time of each layer - ON")
endif()

if(BUILD_X86)
# build x86_ref so
LIST(APPEND X86_LST ${NN2_SRCS} ${REF_SRCS})
add_library(x86_share SHARED ${X86_LST})
SET_TARGET_PROPERTIES(x86_share PROPERTIES OUTPUT_NAME "csi_nn2_ref_x86")
set(X86_BUILD_FLAGS -DCSI_AVX_OPT -DCSI_BUILD_REF -mavx -mfma -fopenmp)
target_compile_options(x86_share PRIVATE ${X86_BUILD_FLAGS})

install(TARGETS x86_share DESTINATION lib)

# build pnna x86 simulate so
LIST(APPEND PNNA_LST ${NN2_SRCS} ${REF_SRCS} ${PNNA_SRCS})
add_library(pnna_share SHARED ${PNNA_LST})
SET_TARGET_PROPERTIES(pnna_share PROPERTIES OUTPUT_NAME "csi_nn2_pnna_x86")
set(PNNA_BUILD_FLAGS -DCSI_BUILD_PNNA)
target_compile_options(pnna_share PRIVATE ${PNNA_BUILD_FLAGS})
target_include_directories(pnna_share PRIVATE module/nna_ddk_install/include/)
set(PNNA_LINK_DIR ${CMAKE_CURRENT_SOURCE_DIR}/module/nna_ddk_install/x86/)
target_link_libraries(pnna_share PRIVATE -L${PNNA_LINK_DIR} -limgdnn_csim -lnnasession_csim)

install(TARGETS pnna_share DESTINATION lib)

# build heterogeneous pnna x86 simulate so
LIST(APPEND HLIGHT_LST ${NN2_SRCS} ${REF_SRCS} ${GREF_SRCS} ${PNNA_SRCS})
add_library(hlight_share SHARED ${HLIGHT_LST})
SET_TARGET_PROPERTIES(hlight_share PROPERTIES OUTPUT_NAME "csi_nn2_hlight_x86")
set(HLIGHT_BUILD_FLAGS -DCSI_BUILD_REF -DCSI_BUILD_GREF -DCSI_BUILD_PNNA)
target_compile_options(hlight_share PRIVATE ${HLIGHT_BUILD_FLAGS})
target_include_directories(hlight_share PRIVATE module/nna_ddk_install/include/)
set(PNNA_LINK_DIR ${CMAKE_CURRENT_SOURCE_DIR}/module/nna_ddk_install/x86/)
target_link_libraries(hlight_share PRIVATE -L${PNNA_LINK_DIR} -limgdnn_csim -lnnasession_csim)

install(TARGETS hlight_share DESTINATION lib)

# build ch8601 so
LIST(APPEND CH8601_LST ${NN2_SRCS} ${CH8601_SRCS})
add_library(ch8601_share SHARED ${CH8601_LST})
SET_TARGET_PROPERTIES(ch8601_share PROPERTIES OUTPUT_NAME "csi_nn2_ch8601")
set(CH8601_BUILD_FLAGS -DCSI_BUILD_REF -DCSI_BUILD_CH8601)
target_compile_options(ch8601_share PRIVATE ${PNNA_BUILD_FLAGS})

install(TARGETS ch8601_share DESTINATION lib)

# build dp1k so
LIST(APPEND DP1K_LST ${NN2_SRCS} ${DP1K_SRCS})
add_library(dp1k_share SHARED ${DP1K_LST})
SET_TARGET_PROPERTIES(dp1k_share PROPERTIES OUTPUT_NAME "csi_nn2_dp1000")
set(DP1K_BUILD_FLAGS -DCSI_BUILD_DP1K -DCSI_BUILD_REF)
target_compile_options(dp1k_share PRIVATE ${DP1K_BUILD_FLAGS})

install(TARGETS dp1k_share DESTINATION lib)
endif()

if(BUILD_RISCV)
# build rvv a
LIST(APPEND RVV_LST ${NN2_SRCS} ${REF_SRCS} ${GREF_SRCS} ${THEAD_RVV_SRCS})
add_library(rvv_static STATIC ${RVV_LST})
SET_TARGET_PROPERTIES(rvv_static PROPERTIES OUTPUT_NAME "csi_nn2_rvv")
set(RVV_BUILD_FLAGS -march=rv64gcv0p7_zfh_xtheadc -mabi=lp64d -DCSI_BUILD_RVV -DCSI_BUILD_REF -DCSI_BUILD_GREF)
target_compile_options(rvv_static PRIVATE ${RVV_BUILD_FLAGS})

install(TARGETS rvv_static DESTINATION lib)

# build c906 a
LIST(APPEND C906_LST ${NN2_SRCS} ${REF_SRCS} ${GREF_SRCS} ${THEAD_RVV_SRCS} ${C906_SRCS})
add_library(c906_static STATIC ${C906_LST})
SET_TARGET_PROPERTIES(c906_static PROPERTIES OUTPUT_NAME "csi_nn2_c906")
set(C906_BUILD_FLAGS -march=rv64gcv0p7_zfh_xtheadc -mabi=lp64d -DCSI_BUILD_C906 -DCSI_BUILD_REF -DCSI_BUILD_GREF)
target_compile_options(c906_static PRIVATE ${C906_BUILD_FLAGS})

install(TARGETS c906_static DESTINATION lib)

# build c908 a
LIST(APPEND C908_LST ${NN2_SRCS} ${REF_SRCS} ${GREF_SRCS} ${THEAD_RVV_SRCS} ${C908_SRCS})
add_library(c908_static STATIC ${C908_LST})
SET_TARGET_PROPERTIES(c908_static PROPERTIES OUTPUT_NAME "csi_nn2_c908")
set(C908_BUILD_FLAGS -march=rv64gcv_zfh_xtheadc_xtheadv -mabi=lp64d -DCSI_BUILD_C908 -DCSI_BUILD_REF -DCSI_BUILD_GREF)
target_compile_options(c908_static PRIVATE ${C908_BUILD_FLAGS})

install(TARGETS c908_static DESTINATION lib)

# build pnna so
LIST(APPEND PNNA_LST ${NN2_SRCS} ${REF_SRCS} ${PNNA_SRCS})
add_library(pnna_share SHARED ${PNNA_LST})
SET_TARGET_PROPERTIES(pnna_share PROPERTIES OUTPUT_NAME "csi_nn2_pnna")
set(PNNA_BUILD_FLAGS -DCSI_BUILD_PNNA)
target_compile_options(pnna_share PRIVATE ${PNNA_BUILD_FLAGS})
target_include_directories(pnna_share PRIVATE module/nna_ddk_install/include/)
set(PNNA_LINK_DIR ${CMAKE_CURRENT_SOURCE_DIR}/module/nna_ddk_install/light/)
target_link_libraries(pnna_share PRIVATE -L${PNNA_LINK_DIR} -limgdnn -lnnasession)

install(TARGETS pnna_share DESTINATION lib)

# build heterogeneous pnna so
LIST(APPEND HLIGHT_LST ${NN2_SRCS} ${REF_SRCS} ${GREF_SRCS} ${PNNA_SRCS})
add_library(hlight_share SHARED ${HLIGHT_LST})
SET_TARGET_PROPERTIES(hlight_share PROPERTIES OUTPUT_NAME "csi_nn2_hlight")
set(HLIGHT_BUILD_FLAGS -DCSI_BUILD_REF -DCSI_BUILD_GREF -DCSI_BUILD_PNNA)
target_compile_options(hlight_share PRIVATE ${HLIGHT_BUILD_FLAGS})
target_include_directories(hlight_share PRIVATE module/nna_ddk_install/include/)
set(PNNA_LINK_DIR ${CMAKE_CURRENT_SOURCE_DIR}/module/nna_ddk_install/light/)
target_link_libraries(hlight_share PRIVATE -L${PNNA_LINK_DIR} -limgdnn -lnnasession)

install(TARGETS hlight_share DESTINATION lib)
endif()

if(BUILD_RISCV_ELF)
# build c906 elf a
LIST(APPEND C906_LST ${NN2_SRCS} ${REF_SRCS} ${GREF_SRCS} ${THEAD_RVV_SRCS} ${C906_SRCS})
add_library(c906_elf_static STATIC ${C906_LST})
SET_TARGET_PROPERTIES(c906_elf_static PROPERTIES OUTPUT_NAME "csi_nn2_c906_rtos")
set(C906_BUILD_FLAGS -march=rv64gcv0p7_zfh_xtheadc -mabi=lp64d -DCSI_BUILD_C906 -DCSI_BUILD_REF -DCSI_BUILD_GREF -DCSI_BUILD_RTOS)
target_compile_options(c906_elf_static PRIVATE ${C906_BUILD_FLAGS})

install(TARGETS c906_elf_static DESTINATION lib)

# build ASP elf a
LIST(APPEND ASP_LST ${NN2_SRCS} ${REF_SRCS} ${GREF_SRCS} ${ASP_SRCS})
add_library(asp_elf_static STATIC ${ASP_LST})
SET_TARGET_PROPERTIES(asp_elf_static PROPERTIES OUTPUT_NAME "csi_nn2_asp")
set(ASP_BUILD_FLAGS -march=rv32imafdcp -mabi=ilp32d -DCSI_BUILD_ASP -DCSI_BUILD_REF -DCSI_BUILD_GREF -DCSI_BUILD_RTOS)
target_compile_options(asp_elf_static PRIVATE ${ASP_BUILD_FLAGS})

install(TARGETS asp_elf_static DESTINATION lib)
endif()

if(BUILD_CSKY)
# build openvx so
LIST(APPEND OPENVX_LST ${NN2_SRCS} ${OPENVX_SRCS})
add_library(openvx_share SHARED ${OPENVX_LST})
SET_TARGET_PROPERTIES(openvx_share PROPERTIES OUTPUT_NAME "csi_nn2_openvx")
set(OPENVX_BUILD_FLAGS -mcpu=c860v -fPIC -DCSI_BUILD_OPENVX -mhard-float)
target_compile_options(openvx_share PRIVATE ${OPENVX_BUILD_FLAGS})
set(OPENVX_LINK_DIR ${CMAKE_CURRENT_SOURCE_DIR}/module/acuity-driver/driver/build/sdk/drivers)
target_link_libraries(openvx_share PRIVATE -mcpu=c860v -fPIC -mhard-float -L${OPENVX_LINK_DIR} -lArchModelSw -lNNArchPerf -lOpenVX -lOpenVXU -lCLC -lVSC -lGAL -lNNGPUBinary -lovxlib -lOvx12VXCBinary)
target_include_directories(openvx_share PRIVATE module/acuity-driver/lib/acuity-ovxlib-dev/include/)
target_include_directories(openvx_share PRIVATE module/acuity-driver/driver/build/sdk/include/)

install(TARGETS openvx_share DESTINATION lib)

# build c860 a
LIST(APPEND C860_LST ${NN2_SRCS} ${REF_SRCS} ${C860_SRCS})
add_library(c860_static STATIC ${C860_LST})
SET_TARGET_PROPERTIES(c860_static PROPERTIES OUTPUT_NAME "csi_nn2_c860")
set(C860_BUILD_FLAGS -mcpu=c860v -DCSI_BUILD_REF)
target_compile_options(c860_static PRIVATE ${C860_BUILD_FLAGS})

install(TARGETS c860_static DESTINATION lib)
endif()

if(BUILD_CSKY_ELF)
# build i805 ref a
LIST(APPEND I805_REF_LST ${NN2_SRCS} ${REF_SRCS} ${I805_REF_SRCS})
add_library(i805_ref_static STATIC ${I805_REF_LST})
SET_TARGET_PROPERTIES(i805_ref_static PROPERTIES OUTPUT_NAME "csi_nn2_ref_i805")
set(I805_REF_BUILD_FLAGS -DCSI_BUILD_REF_I805 -DCSI_MATH_DSP -DCSI_BUILD_RTOS -mcpu=i805)
target_compile_options(i805_ref_static PRIVATE ${I805_REF_BUILD_FLAGS})
target_include_directories(i805_ref_static PRIVATE include/include_xt800)

install(TARGETS i805_ref_static DESTINATION lib)

# build i805 a
LIST(APPEND I805_LST ${NN2_SRCS} ${REF_SRCS} ${I805_SRCS})
add_library(i805_static STATIC ${I805_LST})
SET_TARGET_PROPERTIES(i805_static PROPERTIES OUTPUT_NAME "csi_nn2_i805")
set(I805_BUILD_FLAGS -DCSI_BUILD_I805 -DCSI_MATH_DSP -DCSI_BUILD_RTOS -mcpu=ck805ef -mhard-float)
target_compile_options(i805_static PRIVATE ${I805_BUILD_FLAGS})
target_include_directories(i805_static PRIVATE include/include_xt800)

install(TARGETS i805_static DESTINATION lib)

# build e804 a
LIST(APPEND E804_LST ${NN2_SRCS} ${REF_SRCS} ${E804_SRCS})
add_library(e804_static STATIC ${E804_LST})
SET_TARGET_PROPERTIES(e804_static PROPERTIES OUTPUT_NAME "csi_nn2_e804")
set(E804_BUILD_FLAGS -DCSI_BUILD_E804 -mcpu=e804d -DCSI_BUILD_RTOS -mno-required-attr-fpu-abi)
target_compile_options(e804_static PRIVATE ${E804_BUILD_FLAGS})
target_include_directories(e804_static PRIVATE include/include_xt800)

install(TARGETS e804_static DESTINATION lib)
endif()

install(DIRECTORY "include/." DESTINATION "include" FILES_MATCHING PATTERN "*.h")
71 changes: 31 additions & 40 deletions Makefile
View file
Original file line number Diff line number Diff line change
@@ -1,61 +1,52 @@
CROSS_COMPILE ?= csky-abiv2-elf-
INSTALL_DIR = ../../lib/
NN2_ROOT := $(shell pwd)
all: nn2_ref_x86

ifeq ($(GCOV),y)
EXTRA_CFLAGS = -fprofile-arcs -ftest-coverage -g -O0
LIBS += -fprofile-arcs -ftest-coverage -lgcov
else
EXTRA_CFLAGS = -O2 -g -Werror -DCSI_DEBUG
endif
nn2_c860:
mkdir -p csky_build; cd csky_build; cmake ../ -DBUILD_CSKY=ON -DCMAKE_BUILD_TYPE=Release; make c860_static -j8; cd -

export CROSS_COMPILE INSTALL_DIR
nn2_rvv:
mkdir -p riscv_build; cd riscv_build; cmake ../ -DBUILD_RISCV=ON -DCMAKE_BUILD_TYPE=Release; make rvv_static -j8; cd -

nn2_c906:
mkdir -p riscv_build; cd riscv_build; cmake ../ -DBUILD_RISCV=ON -DCMAKE_BUILD_TYPE=Release; make c906_static -j8; cd -

all: nn2_ref_x86
nn2_c906_elf:
mkdir -p riscv_elf_build; cd riscv_elf_build; cmake ../ -DBUILD_RISCV_ELF=ON -DCMAKE_BUILD_TYPE=Release; make c906_elf_static -j8; cd -

nn2_c860:
DSP_LIB="libcsi_nn2_c860" CFLAGS="-mcpu=c860v -DCSI_BUILD_REF $(EXTRA_CFLAGS)" \
CROSS_COMPILE="csky-abiv2-linux-" NN2_ROOT=${NN2_ROOT} make -C build_script/nn2_c860 -j8
cd source/; find . -name *.o | xargs rm; cd -
nn2_asp_elf:
mkdir -p riscv_elf_build; cd riscv_elf_build; cmake ../ -DBUILD_RISCV_ELF=ON -DCMAKE_BUILD_TYPE=Release; make asp_elf_static -j8; cd -

nn2_c906:
DSP_LIB="libcsi_nn2_c906" CFLAGS="-march=rv64gcvxthead -mabi=lp64dv -DCSI_BUILD_C906 -DCSI_BUILD_REF -DCSI_BUILD_GREF $(EXTRA_CFLAGS)" \
CROSS_COMPILE="riscv64-unknown-linux-gnu-" NN2_ROOT=${NN2_ROOT} make -C build_script/nn2_c906 -j8
cd source/; find . -name *.o | xargs rm; cd -
nn2_c908:
mkdir -p riscv_build; cd riscv_build; cmake ../ -DBUILD_RISCV=ON -DCMAKE_BUILD_TYPE=Release; make c908_static -j8; cd -

nn2_ref_x86:
DSP_LIB="libcsi_nn2_ref_x86" CFLAGS="$(EXTRA_CFLAGS) -DCSI_BUILD_REF -fPIC -DCSI_AVX_OPT -mavx -mfma -fopenmp" \
CROSS_COMPILE="" NN2_ROOT=${NN2_ROOT} make -C build_script/nn2_ref -j8
cd source/; find . -name *.o | xargs rm; cd -
DSP_LIB="libcsi_nn2_ref_x86" CFLAGS="$(EXTRA_CFLAGS) -DCSI_BUILD_REF -fPIC -DCSI_AVX_OPT -mavx -mfma -fopenmp" \
CROSS_COMPILE="" NN2_ROOT=${NN2_ROOT} make -C build_script/nn2_ref nn2_shared -j8
cd source/; find . -name *.o | xargs rm; cd -
mkdir -p x86_build; cd x86_build; cmake ../ -DBUILD_X86=ON -DCMAKE_BUILD_TYPE=Release; make x86_share -j8; cd -

nn2_openvx:
mkdir -p csky_build; cd csky_build; cmake ../ -DBUILD_CSKY=ON -DCMAKE_BUILD_TYPE=Release; make openvx_share -j8; cd -

nn2_ref_i805:
DSP_LIB="libcsi_nn2_ref_i805.a" CFLAGS="-DCSI_BUILD_REF_I805 -DCSI_MATH_DSP -mcpu=i805 $(EXTRA_CFLAGS)" \
CROSS_COMPILE="csky-abiv2-elf-" NN2_ROOT=${NN2_ROOT} make -C build_script/nn2_ref_i805 -j8
cd source/; find . -name *.o | xargs rm; cd -
nn2_pnna:
mkdir -p riscv_build; cd riscv_build; cmake ../ -DBUILD_RISCV=ON -DCMAKE_BUILD_TYPE=Release; make pnna_share -j8; cd -

nn2_e804:
DSP_LIB="libcsi_nn2_e804.a" CFLAGS="-DCSI_BUILD_E804 -mcpu=e804d -mno-required-attr-fpu-abi $(EXTRA_CFLAGS)" \
CROSS_COMPILE="csky-abiv2-elf-" NN2_ROOT=${NN2_ROOT} make -C build_script/nn2_e804 -j8
cd source/; find . -name *.o | xargs rm; cd -
nn2_pnna_x86:
mkdir -p x86_build; cd x86_build; cmake ../ -DBUILD_X86=ON -DCMAKE_BUILD_TYPE=Release; make pnna_share -j8; cd -

nn2_i805:
DSP_LIB="libcsi_nn2_i805.a" CFLAGS="-DCSI_BUILD_I805 -DCSI_BUILD_REF -DCSI_BUILD_GREF -mcpu=ck805ef -mhard-float $(EXTRA_CFLAGS)" \
CROSS_COMPILE="csky-abiv2-elf-" NN2_ROOT=${NN2_ROOT} make -C build_script/nn2_i805 -j8
cd source/; find . -name *.o | xargs rm; cd -
nn2_hlight_x86:
mkdir -p x86_build; cd x86_build; cmake ../ -DBUILD_X86=ON -DCMAKE_BUILD_TYPE=Release; make hlight_share -j8; cd -

nn2_hlight:
mkdir -p riscv_build; cd riscv_build; cmake ../ -DBUILD_RISCV=ON -DCMAKE_BUILD_TYPE=Release; make hlight_share -j8; cd -

.PHONY: install_nn2
install_nn2: include
mkdir -p install_nn2/lib
cp include install_nn2 -r
cp lib/libcsi_nn2_* install_nn2/lib -rf
-cp riscv_build/libcsi_nn2_* install_nn2/lib -rf
-cp csky_build/libcsi_nn2_* install_nn2/lib -rf
-cp x86_build/libcsi_nn2_* install_nn2/lib -rf
cp version install_nn2/ -rf

clint:
./script/git-clang-format.sh origin/master

clean:
rm lib/* -rf
find . -name *.o | xargs rm -rf
rm lib/* -rf
Loading

0 comments on commit 0e784c3

Please sign in to comment.