diff --git a/tensor_computing/src/CMakeLists.txt b/tensor_computing/src/CMakeLists.txt index 55e59b9a..d04ddc78 100755 --- a/tensor_computing/src/CMakeLists.txt +++ b/tensor_computing/src/CMakeLists.txt @@ -1,10 +1,11 @@ file(GLOB general_srcs ${CMAKE_CURRENT_SOURCE_DIR}/cpu/general/*.cpp) if (USE_NEON) - file(GLOB arm_fp16_srcs ${CMAKE_CURRENT_SOURCE_DIR}/cpu/arm/fp16/*cpp) + file(GLOB arm_srcs ${CMAKE_CURRENT_SOURCE_DIR}/cpu/arm/*.cpp) + file(GLOB arm_fp16_srcs ${CMAKE_CURRENT_SOURCE_DIR}/cpu/arm/fp16/*.cpp) if (USE_INT8) - file(GLOB arm_int8_srcs ${CMAKE_CURRENT_SOURCE_DIR}/cpu/arm/int8/*cpp) + file(GLOB arm_int8_srcs ${CMAKE_CURRENT_SOURCE_DIR}/cpu/arm/int8/*.cpp) endif (USE_INT8) - file(GLOB arm_bnn_srcs ${CMAKE_CURRENT_SOURCE_DIR}/cpu/arm/*.cpp) + file(GLOB arm_bnn_srcs ${CMAKE_CURRENT_SOURCE_DIR}/cpu/arm/bnn/*.cpp) set(arm_srcs "${arm_srcs};${arm_fp16_srcs};${arm_int8_srcs};${arm_bnn_srcs}") endif (USE_NEON)