Merge pull request #9 from songqun/master

tensor_computing: fix compiling bug that conv_bnn cannot be ld
huawei-noah · Dec 15, 2019 · 62609ca · 62609ca
2 parents 4e5298e + 356cbde
commit 62609ca
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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)