You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
- project(optimize)
-
- set(NNACL_DIR ${CMAKE_CURRENT_SOURCE_DIR}/..)
- include_directories(NNACL_DIR)
-
- ########################### optimized files ###########################
- file(GLOB SDOT_SRC ${NNACL_DIR}/assembly/opt/*.S)
- file(GLOB FP16_C_SRC ${NNACL_DIR}/fp16/*.c)
- file(GLOB FP16_NEON_SRC ${NNACL_DIR}/assembly/fp16/*.S)
-
- set_property(SOURCE ${SDOT_SRC} PROPERTY LANGUAGE C)
- set_property(SOURCE ${FP16_C_SRC} PROPERTY LANGUAGE C)
- set_property(SOURCE ${FP16_NEON_SRC} PROPERTY LANGUAGE C)
-
- ########################### share library build ########################
- list(APPEND SDOT_FILES ${SDOT_SRC})
- list(APPEND FP16_FILES ${FP16_C_SRC})
- list(APPEND FP16_FILES ${FP16_NEON_SRC})
-
- string(REPLACE "-fvisibility=hidden" "-fvisibility=default" CMAKE_C_FLAGS "${CMAKE_C_FLAGS}")
- set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -march=armv8.2-a+dotprod+fp16")
- set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=armv8.2-a+dotprod+fp16")
-
- add_library(nnacl_optimize STATIC ${SDOT_FILES})
- target_link_libraries(nnacl_optimize mindspore-lite)
-
- add_library(nnacl_fp16 STATIC ${FP16_FILES})
- target_link_libraries(nnacl_fp16 mindspore-lite)
|
