nihui
/
ncnn
Not watched
1
0
Fork 0
Code
Releases 46 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
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.
Tree: 8704ed418e
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '8704ed418e'
${ noResults }
ncnn/cmake/ncnn_add_layer.cmake

230 lines
15 kB
Raw Blame History 

  1. 

  2. macro(ncnn_add_layer class)

  3.     string(TOLOWER ${class} name)

  4. 

  5.     # WITH_LAYER_xxx option

  6.     if(${ARGC} EQUAL 2)

  7.         option(WITH_LAYER_${name} "build with layer ${name}" ${ARGV1})

  8.     else()

  9.         option(WITH_LAYER_${name} "build with layer ${name}" ON)

  10.     endif()

  11. 

  12.     if(NCNN_CMAKE_VERBOSE)

  13.         message(STATUS "WITH_LAYER_${name} = ${WITH_LAYER_${name}}")

  14.     endif()

  15. 

  16.     if(WITH_LAYER_${name})

  17.         list(APPEND ncnn_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/layer/${name}.cpp)

  18. 

  19.         # look for arch specific implementation and append source

  20.         # optimized implementation for armv7, aarch64 or x86

  21.         set(LAYER_ARCH_SRC ${CMAKE_CURRENT_SOURCE_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}.cpp)

  22.         if(EXISTS ${LAYER_ARCH_SRC})

  23.             set(WITH_LAYER_${name}_${NCNN_TARGET_ARCH} 1)

  24.             list(APPEND ncnn_SRCS ${LAYER_ARCH_SRC})

  25.         endif()

  26. 

  27.         set(LAYER_VULKAN_SRC ${CMAKE_CURRENT_SOURCE_DIR}/layer/vulkan/${name}_vulkan.cpp)

  28.         if(NCNN_VULKAN AND EXISTS ${LAYER_VULKAN_SRC})

  29.             set(WITH_LAYER_${name}_vulkan 1)

  30.             list(APPEND ncnn_SRCS ${LAYER_VULKAN_SRC})

  31.         endif()

  32.     endif()

  33. 

  34.     # generate layer_declaration and layer_registry file

  35.     if(WITH_LAYER_${name})

  36.         set(layer_declaration "${layer_declaration}#include \"layer/${name}.h\"\n")

  37.         set(layer_declaration_class "class ${class}_final : virtual public ${class}")

  38.         set(create_pipeline_content "        { int ret = ${class}::create_pipeline(opt); if (ret) return ret; }\n")

  39.         set(destroy_pipeline_content "        { int ret = ${class}::destroy_pipeline(opt); if (ret) return ret; }\n")

  40. 

  41.         source_group ("sources\\\\layers" FILES "${CMAKE_CURRENT_SOURCE_DIR}/layer/${name}.cpp")

  42.     endif()

  43. 

  44.     if(WITH_LAYER_${name}_${NCNN_TARGET_ARCH})

  45.         set(layer_declaration "${layer_declaration}#include \"layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}.h\"\n")

  46.         set(layer_declaration_class "${layer_declaration_class}, virtual public ${class}_${NCNN_TARGET_ARCH}")

  47.         set(create_pipeline_content "${create_pipeline_content}        { int ret = ${class}_${NCNN_TARGET_ARCH}::create_pipeline(opt); if (ret) return ret; }\n")

  48.         set(destroy_pipeline_content "        { int ret = ${class}_${NCNN_TARGET_ARCH}::destroy_pipeline(opt); if (ret) return ret; }\n${destroy_pipeline_content}")

  49. 

  50.         source_group ("sources\\\\layers\\\\${NCNN_TARGET_ARCH}" FILES "${CMAKE_CURRENT_SOURCE_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}.cpp")

  51.     endif()

  52. 

  53.     if(WITH_LAYER_${name}_vulkan)

  54.         set(layer_declaration "${layer_declaration}#include \"layer/vulkan/${name}_vulkan.h\"\n")

  55.         set(layer_declaration_class "${layer_declaration_class}, virtual public ${class}_vulkan")

  56.         set(create_pipeline_content "${create_pipeline_content}        if (vkdev) { int ret = ${class}_vulkan::create_pipeline(opt); if (ret) return ret; }\n")

  57.         set(destroy_pipeline_content "        if (vkdev) { int ret = ${class}_vulkan::destroy_pipeline(opt); if (ret) return ret; }\n${destroy_pipeline_content}")

  58. 

  59.         file(GLOB_RECURSE NCNN_SHADER_SRCS "layer/vulkan/shader/${name}.comp")

  60.         file(GLOB_RECURSE NCNN_SHADER_SUBSRCS "layer/vulkan/shader/${name}_*.comp")

  61.         list(APPEND NCNN_SHADER_SRCS ${NCNN_SHADER_SUBSRCS})

  62.         foreach(NCNN_SHADER_SRC ${NCNN_SHADER_SRCS})

  63.             ncnn_add_shader(${NCNN_SHADER_SRC})

  64.         endforeach()

  65. 

  66.         source_group ("sources\\\\layers\\\\vulkan" FILES "${CMAKE_CURRENT_SOURCE_DIR}/layer/vulkan/${name}_vulkan.cpp")

  67.     endif()

  68. 

  69.     if(WITH_LAYER_${name})

  70.         set(layer_declaration "${layer_declaration}namespace ncnn {\n${layer_declaration_class}\n{\n")

  71.         set(layer_declaration "${layer_declaration}public:\n")

  72.         set(layer_declaration "${layer_declaration}    virtual int create_pipeline(const Option& opt) {\n${create_pipeline_content}        return 0;\n    }\n")

  73.         set(layer_declaration "${layer_declaration}    virtual int destroy_pipeline(const Option& opt) {\n${destroy_pipeline_content}        return 0;\n    }\n")

  74.         set(layer_declaration "${layer_declaration}};\n")

  75.         set(layer_declaration "${layer_declaration}DEFINE_LAYER_CREATOR(${class}_final)\n} // namespace ncnn\n\n")

  76.     endif()

  77. 

  78.     if(WITH_LAYER_${name})

  79.         set(layer_registry "${layer_registry}#if NCNN_STRING\n{\"${class}\",${class}_final_layer_creator},\n#else\n{${class}_final_layer_creator},\n#endif\n")

  80.     else()

  81.         set(layer_registry "${layer_registry}#if NCNN_STRING\n{\"${class}\",0},\n#else\n{0},\n#endif\n")

  82.     endif()

  83. 

  84. 

  85.     if(NCNN_RUNTIME_CPU AND NCNN_AVX2 AND NCNN_TARGET_ARCH STREQUAL "x86")

  86.         # enable avx2

  87.         set(NCNN_X86_HEADER ${CMAKE_CURRENT_SOURCE_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}.h)

  88.         set(NCNN_X86_SOURCE ${CMAKE_CURRENT_SOURCE_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}.cpp)

  89. 

  90.         if(WITH_LAYER_${name} AND EXISTS ${NCNN_X86_HEADER} AND EXISTS ${NCNN_X86_SOURCE})

  91. 

  92.             set(NCNN_AVX2_HEADER ${CMAKE_CURRENT_BINARY_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}_avx2.h)

  93.             set(NCNN_AVX2_SOURCE ${CMAKE_CURRENT_BINARY_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}_avx2.cpp)

  94. 

  95.             add_custom_command(

  96.                 OUTPUT ${NCNN_AVX2_HEADER}

  97.                 COMMAND ${CMAKE_COMMAND} -DSRC=${NCNN_X86_HEADER} -DDST=${NCNN_AVX2_HEADER} -DCLASS=${class} -P "${CMAKE_CURRENT_SOURCE_DIR}/../cmake/ncnn_generate_avx2_source.cmake"

  98.                 DEPENDS ${NCNN_X86_HEADER}

  99.                 COMMENT "Generating source ${name}_${NCNN_TARGET_ARCH}_avx2.h"

  100.                 VERBATIM

  101.             )

  102.             set_source_files_properties(${NCNN_AVX2_HEADER} PROPERTIES GENERATED TRUE)

  103. 

  104.             add_custom_command(

  105.                 OUTPUT ${NCNN_AVX2_SOURCE}

  106.                 COMMAND ${CMAKE_COMMAND} -DSRC=${NCNN_X86_SOURCE} -DDST=${NCNN_AVX2_SOURCE} -DCLASS=${class} -P "${CMAKE_CURRENT_SOURCE_DIR}/../cmake/ncnn_generate_avx2_source.cmake"

  107.                 DEPENDS ${NCNN_X86_SOURCE}

  108.                 COMMENT "Generating source ${name}_${NCNN_TARGET_ARCH}_avx2.cpp"

  109.                 VERBATIM

  110.             )

  111.             set_source_files_properties(${NCNN_AVX2_SOURCE} PROPERTIES GENERATED TRUE)

  112. 

  113.             if(CMAKE_CXX_COMPILER_ID MATCHES "MSVC" OR (CMAKE_CXX_COMPILER_ID MATCHES "Clang" AND CMAKE_CXX_SIMULATE_ID MATCHES "MSVC"))

  114.                 set_source_files_properties(${NCNN_AVX2_SOURCE} PROPERTIES COMPILE_FLAGS "/arch:AVX2 /DAVX2 /fp:strict")

  115.             else()

  116.                 set_source_files_properties(${NCNN_AVX2_SOURCE} PROPERTIES COMPILE_FLAGS "-mfma -mf16c -mavx2")

  117.             endif()

  118. 

  119.             list(APPEND ncnn_SRCS ${NCNN_AVX2_HEADER} ${NCNN_AVX2_SOURCE})

  120. 

  121.             # generate layer_declaration and layer_registry_avx2 file

  122.             set(layer_declaration "${layer_declaration}#include \"layer/${name}.h\"\n")

  123.             set(layer_declaration_class "class ${class}_final_avx2 : virtual public ${class}")

  124.             set(create_pipeline_content "        { int ret = ${class}::create_pipeline(opt); if (ret) return ret; }\n")

  125.             set(destroy_pipeline_content "        { int ret = ${class}::destroy_pipeline(opt); if (ret) return ret; }\n")

  126. 

  127.             set(layer_declaration "${layer_declaration}#include \"layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}_avx2.h\"\n")

  128.             set(layer_declaration_class "${layer_declaration_class}, virtual public ${class}_${NCNN_TARGET_ARCH}_avx2")

  129.             set(create_pipeline_content "${create_pipeline_content}        { int ret = ${class}_${NCNN_TARGET_ARCH}_avx2::create_pipeline(opt); if (ret) return ret; }\n")

  130.             set(destroy_pipeline_content "        { int ret = ${class}_${NCNN_TARGET_ARCH}_avx2::destroy_pipeline(opt); if (ret) return ret; }\n${destroy_pipeline_content}")

  131. 

  132.             if(WITH_LAYER_${name}_vulkan)

  133.                 set(layer_declaration "${layer_declaration}#include \"layer/vulkan/${name}_vulkan.h\"\n")

  134.                 set(layer_declaration_class "${layer_declaration_class}, virtual public ${class}_vulkan")

  135.                 set(create_pipeline_content "${create_pipeline_content}        if (vkdev) { int ret = ${class}_vulkan::create_pipeline(opt); if (ret) return ret; }\n")

  136.                 set(destroy_pipeline_content "        if (vkdev) { int ret = ${class}_vulkan::destroy_pipeline(opt); if (ret) return ret; }\n${destroy_pipeline_content}")

  137.             endif()

  138. 

  139.             set(layer_declaration "${layer_declaration}namespace ncnn {\n${layer_declaration_class}\n{\n")

  140.             set(layer_declaration "${layer_declaration}public:\n")

  141.             set(layer_declaration "${layer_declaration}    virtual int create_pipeline(const Option& opt) {\n${create_pipeline_content}        return 0;\n    }\n")

  142.             set(layer_declaration "${layer_declaration}    virtual int destroy_pipeline(const Option& opt) {\n${destroy_pipeline_content}        return 0;\n    }\n")

  143.             set(layer_declaration "${layer_declaration}};\n")

  144.             set(layer_declaration "${layer_declaration}DEFINE_LAYER_CREATOR(${class}_final_avx2)\n} // namespace ncnn\n\n")

  145. 

  146.             set(layer_registry_avx2 "${layer_registry_avx2}#if NCNN_STRING\n{\"${class}\",${class}_final_avx2_layer_creator},\n#else\n{${class}_final_avx2_layer_creator},\n#endif\n")

  147.         else()

  148.             # no arm optimized version

  149.             if(WITH_LAYER_${name})

  150.                 set(layer_registry_avx2 "${layer_registry_avx2}#if NCNN_STRING\n{\"${class}\",${class}_final_layer_creator},\n#else\n{${class}_final_layer_creator},\n#endif\n")

  151.             else()

  152.                 set(layer_registry_avx2 "${layer_registry_avx2}#if NCNN_STRING\n{\"${class}\",0},\n#else\n{0},\n#endif\n")

  153.             endif()

  154.         endif()

  155.     endif()

  156. 

  157.     if(NCNN_RUNTIME_CPU AND NCNN_ARM82 AND ((IOS AND CMAKE_OSX_ARCHITECTURES MATCHES "arm64") OR (CMAKE_SYSTEM_PROCESSOR MATCHES "^(aarch64)")))

  158.         # enable armv8.2a+fp16

  159.         set(NCNN_ARM_HEADER ${CMAKE_CURRENT_SOURCE_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}.h)

  160.         set(NCNN_ARM_SOURCE ${CMAKE_CURRENT_SOURCE_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}.cpp)

  161. 

  162.         if(WITH_LAYER_${name} AND EXISTS ${NCNN_ARM_HEADER} AND EXISTS ${NCNN_ARM_SOURCE})

  163. 

  164.             set(NCNN_ARM82_HEADER ${CMAKE_CURRENT_BINARY_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}_arm82.h)

  165.             set(NCNN_ARM82_SOURCE ${CMAKE_CURRENT_BINARY_DIR}/layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}_arm82.cpp)

  166. 

  167.             add_custom_command(

  168.                 OUTPUT ${NCNN_ARM82_HEADER}

  169.                 COMMAND ${CMAKE_COMMAND} -DSRC=${NCNN_ARM_HEADER} -DDST=${NCNN_ARM82_HEADER} -DCLASS=${class} -P "${CMAKE_CURRENT_SOURCE_DIR}/../cmake/ncnn_generate_arm82_source.cmake"

  170.                 DEPENDS ${NCNN_ARM_HEADER}

  171.                 COMMENT "Generating source ${name}_${NCNN_TARGET_ARCH}_arm82.h"

  172.                 VERBATIM

  173.             )

  174.             set_source_files_properties(${NCNN_ARM82_HEADER} PROPERTIES GENERATED TRUE)

  175. 

  176.             add_custom_command(

  177.                 OUTPUT ${NCNN_ARM82_SOURCE}

  178.                 COMMAND ${CMAKE_COMMAND} -DSRC=${NCNN_ARM_SOURCE} -DDST=${NCNN_ARM82_SOURCE} -DCLASS=${class} -P "${CMAKE_CURRENT_SOURCE_DIR}/../cmake/ncnn_generate_arm82_source.cmake"

  179.                 DEPENDS ${NCNN_ARM_SOURCE}

  180.                 COMMENT "Generating source ${name}_${NCNN_TARGET_ARCH}_arm82.cpp"

  181.                 VERBATIM

  182.             )

  183.             set_source_files_properties(${NCNN_ARM82_SOURCE} PROPERTIES GENERATED TRUE)

  184. 

  185.             set_source_files_properties(${NCNN_ARM82_SOURCE} PROPERTIES COMPILE_FLAGS "-march=armv8.2-a+fp16")

  186. 

  187.             list(APPEND ncnn_SRCS ${NCNN_ARM82_HEADER} ${NCNN_ARM82_SOURCE})

  188. 

  189.             # generate layer_declaration and layer_registry_arm82 file

  190.             set(layer_declaration "${layer_declaration}#include \"layer/${name}.h\"\n")

  191.             set(layer_declaration_class "class ${class}_final_arm82 : virtual public ${class}")

  192.             set(create_pipeline_content "        { int ret = ${class}::create_pipeline(opt); if (ret) return ret; }\n")

  193.             set(destroy_pipeline_content "        { int ret = ${class}::destroy_pipeline(opt); if (ret) return ret; }\n")

  194. 

  195.             set(layer_declaration "${layer_declaration}#include \"layer/${NCNN_TARGET_ARCH}/${name}_${NCNN_TARGET_ARCH}_arm82.h\"\n")

  196.             set(layer_declaration_class "${layer_declaration_class}, virtual public ${class}_${NCNN_TARGET_ARCH}_arm82")

  197.             set(create_pipeline_content "${create_pipeline_content}        { int ret = ${class}_${NCNN_TARGET_ARCH}_arm82::create_pipeline(opt); if (ret) return ret; }\n")

  198.             set(destroy_pipeline_content "        { int ret = ${class}_${NCNN_TARGET_ARCH}_arm82::destroy_pipeline(opt); if (ret) return ret; }\n${destroy_pipeline_content}")

  199. 

  200.             if(WITH_LAYER_${name}_vulkan)

  201.                 set(layer_declaration "${layer_declaration}#include \"layer/vulkan/${name}_vulkan.h\"\n")

  202.                 set(layer_declaration_class "${layer_declaration_class}, virtual public ${class}_vulkan")

  203.                 set(create_pipeline_content "${create_pipeline_content}        if (vkdev) { int ret = ${class}_vulkan::create_pipeline(opt); if (ret) return ret; }\n")

  204.                 set(destroy_pipeline_content "        if (vkdev) { int ret = ${class}_vulkan::destroy_pipeline(opt); if (ret) return ret; }\n${destroy_pipeline_content}")

  205.             endif()

  206. 

  207.             set(layer_declaration "${layer_declaration}namespace ncnn {\n${layer_declaration_class}\n{\n")

  208.             set(layer_declaration "${layer_declaration}public:\n")

  209.             set(layer_declaration "${layer_declaration}    virtual int create_pipeline(const Option& opt) {\n${create_pipeline_content}        return 0;\n    }\n")

  210.             set(layer_declaration "${layer_declaration}    virtual int destroy_pipeline(const Option& opt) {\n${destroy_pipeline_content}        return 0;\n    }\n")

  211.             set(layer_declaration "${layer_declaration}};\n")

  212.             set(layer_declaration "${layer_declaration}DEFINE_LAYER_CREATOR(${class}_final_arm82)\n} // namespace ncnn\n\n")

  213. 

  214.             set(layer_registry_arm82 "${layer_registry_arm82}#if NCNN_STRING\n{\"${class}\",${class}_final_arm82_layer_creator},\n#else\n{${class}_final_arm82_layer_creator},\n#endif\n")

  215.         else()

  216.             # no arm optimized version

  217.             if(WITH_LAYER_${name})

  218.                 set(layer_registry_arm82 "${layer_registry_arm82}#if NCNN_STRING\n{\"${class}\",${class}_final_layer_creator},\n#else\n{${class}_final_layer_creator},\n#endif\n")

  219.             else()

  220.                 set(layer_registry_arm82 "${layer_registry_arm82}#if NCNN_STRING\n{\"${class}\",0},\n#else\n{0},\n#endif\n")

  221.             endif()

  222.         endif()

  223.     endif()

  224. 

  225.     # generate layer_type_enum file

  226.     set(layer_type_enum "${layer_type_enum}${class} = ${__LAYER_TYPE_ENUM_INDEX},\n")

  227.     math(EXPR __LAYER_TYPE_ENUM_INDEX "${__LAYER_TYPE_ENUM_INDEX}+1")

  228. endmacro()