OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
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: 6e882c1a86
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '6e882c1a86'
${ noResults }
MegEngine/imperative/CMakeLists.txt

119 lines
6.5 kB
Raw Blame History 

  1. find_package(NumPy REQUIRED)

  2. 

  3. set(PACKAGE_NAME megengine)

  4. set(PACKAGE_NAME ${PACKAGE_NAME} PARENT_SCOPE)

  5. set(MODULE_NAME _imperative_rt)

  6. set(MODULE_NAME ${MODULE_NAME} PARENT_SCOPE)

  7. file(GLOB_RECURSE SRCS src/impl/*.cpp src/include/*.h python/src/*.cpp python/src/*.h)

  8. 

  9. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DMGB_WITH_IMPERATIVE=1")

  10. 

  11. file(GLOB_RECURSE OPR_DECL_SRCS "${PROJECT_SOURCE_DIR}/src/**/*.oprdecl")

  12. file(GLOB_RECURSE PYTHON_SRCS python/${PACKAGE_NAME}/*.py)

  13. list(REMOVE_ITEM PYTHON_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/python/megengine/core/ops/_internal/generated_ops.py ${CMAKE_CURRENT_SOURCE_DIR}/python/megengine/core/ops/_internal/param_defs.py)

  14. file(GLOB_RECURSE ALL_HEADERS src/cpp/megbrain_pubapi.h

  15.     ${PROJECT_SOURCE_DIR}/src/core/include/*

  16.     ${PROJECT_SOURCE_DIR}/src/opr/include/*

  17.     ${PROJECT_SOURCE_DIR}/src/serialization/include/*

  18.     ${PROJECT_SOURCE_DIR}/src/plugin/include/*

  19.     ${PROJECT_SOURCE_DIR}/dnn/include/*)

  20. 

  21. set(MEGENGINE_DIR ${CMAKE_CURRENT_BINARY_DIR}/python/)

  22. set(GEN_OPS_DIR ${MEGENGINE_DIR}/${PACKAGE_NAME}/core/ops/_internal)

  23. file(MAKE_DIRECTORY ${GEN_OPS_DIR})

  24. set(GEN_OPS_FILE ${GEN_OPS_DIR}/generated_ops.py)

  25. set(GEN_OP_PARAMS_FILE ${MEGENGINE_DIR}/${PACKAGE_NAME}/core/ops/_internal/param_defs.py)

  26. set(GEN_OP_PARAMS_TEMPLATE ${CMAKE_CURRENT_SOURCE_DIR}/python/tools/ops.tpl.py)

  27. 

  28. ##################### generate python opr_param_defs.py   ##############

  29. 

  30. file(COPY ${PROJECT_SOURCE_DIR}/dnn/scripts/opr_param_defs.py DESTINATION ${CMAKE_CURRENT_BINARY_DIR})

  31. file(READ ${PROJECT_SOURCE_DIR}/tools/param_defs/mgb_opr_param_defs.py CONTENTS)

  32. file(APPEND ${CMAKE_CURRENT_BINARY_DIR}/opr_param_defs.py ${CONTENTS})

  33. 

  34. add_custom_command(

  35.     OUTPUT ${GEN_OPS_FILE}

  36.     COMMAND ${CMAKE_COMMAND} -E touch ${MEGENGINE_DIR}/${PACKAGE_NAME}/core/${MODULE_NAME}.so ${GEN_OPS_FILE} ${GEN_OP_PARAMS_FILE}

  37.     COMMAND ${CMAKE_COMMAND} -E copy_directory ${CMAKE_CURRENT_SOURCE_DIR}/python/${PACKAGE_NAME} ${MEGENGINE_DIR}/${PACKAGE_NAME}

  38.     COMMAND ${CMAKE_COMMAND} -E remove -f ${MEGENGINE_DIR}/${PACKAGE_NAME}/core/${MODULE_NAME}.so ${GEN_OPS_FILE} ${GEN_OP_PARAMS_FILE}

  39.     COMMAND ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/python/tools/gen_ops.py ${OPR_DECL_SRCS} -o ${GEN_OPS_FILE}

  40.     COMMAND ${CMAKE_COMMAND} -E copy_directory ${CMAKE_CURRENT_SOURCE_DIR}/python/test ${MEGENGINE_DIR}/${PACKAGE_NAME}/test

  41.     COMMAND ${PYTHON_EXECUTABLE} ${PROJECT_SOURCE_DIR}/dnn/scripts/gen_param_defs.py -t py --imperative ${CMAKE_CURRENT_BINARY_DIR}/opr_param_defs.py ${GEN_OP_PARAMS_FILE}

  42.     DEPENDS ${OPR_DECL_SRCS} ${PYTHON_SRCS} ${ALL_HEADERS} ${GEN_OP_PARAMS_TEMPLATE}

  43.     VERBATIM

  44. )

  45. 

  46. add_custom_target(gen_opr_py DEPENDS ${GEN_OPS_FILE})

  47. 

  48. ##################### generate opdef c header and python binding ##############

  49. 

  50. set(OP_DEF_HEADER_OUT_DIR ${CMAKE_CURRENT_BINARY_DIR}/src/include)

  51. file(MAKE_DIRECTORY ${OP_DEF_HEADER_OUT_DIR}/megbrain/imperative/opdef)

  52. set(OP_DEF_HEADER ${OP_DEF_HEADER_OUT_DIR}/megbrain/imperative/opdef/all.h)

  53. set(OP_DEF_PYTHON_BINDING_OUT_DIR ${MEGENGINE_DIR}/${PACKAGE_NAME}/src)

  54. file(MAKE_DIRECTORY ${OP_DEF_PYTHON_BINDING_OUT_DIR})

  55. set(OP_DEF_PYTHON_BINDING ${OP_DEF_PYTHON_BINDING_OUT_DIR}/opdef.inl)

  56. set(OP_PARAM_DEF ${CMAKE_CURRENT_BINARY_DIR}/opr_param_defs.py)

  57. set(GEN_OP_DEF_SCRIPT ${CMAKE_CURRENT_SOURCE_DIR}/python/tools/gen_op_defs.py)

  58. 

  59. add_custom_command(

  60.     OUTPUT ${OP_DEF_HEADER} ${OP_DEF_PYTHON_BINDING}

  61.     COMMAND ${PYTHON_EXECUTABLE} ${GEN_OP_DEF_SCRIPT} ${OP_PARAM_DEF} ${OP_DEF_HEADER}

  62.     COMMAND ${PYTHON_EXECUTABLE} ${GEN_OP_DEF_SCRIPT} -t py ${OP_PARAM_DEF} ${OP_DEF_PYTHON_BINDING}

  63.     DEPENDS ${GEN_OP_DEF_SCRIPT} ${OP_PARAM_DEF}

  64.     VERBATIM

  65. )

  66. 

  67. add_custom_target(gen_op_def_internal DEPENDS ${OP_DEF_HEADER} ${OP_DEF_PYTHON_BINDING})

  68. add_library(gen_op_def INTERFACE)

  69. target_include_directories(gen_op_def INTERFACE ${OP_DEF_HEADER_OUT_DIR} ${OP_DEF_PYTHON_BINDING_OUT_DIR})

  70. add_dependencies(gen_op_def gen_op_def_internal)

  71. 

  72. ##################### end of opdef generation #########################

  73. 

  74. set(VERSION_SCRIPT ${CMAKE_CURRENT_SOURCE_DIR}/src/version.ld)

  75. add_custom_target(_version_ld SOURCES ${VERSION_SCRIPT})

  76. 

  77. add_subdirectory(${PROJECT_SOURCE_DIR}/third_party/pybind11 ${PROJECT_BINARY_DIR}/third_party/pybind11)

  78. pybind11_add_module(${MODULE_NAME} NO_EXTRAS ${SRCS})

  79. if (APPLE OR MSVC OR WIN32)

  80.     target_link_libraries(${MODULE_NAME} PRIVATE gen_op_def megbrain megdnn)

  81. else()

  82.     target_link_libraries(${MODULE_NAME} PRIVATE gen_op_def megbrain megdnn -Wl,--version-script=${VERSION_SCRIPT})

  83. endif()

  84. if (MGE_WITH_DISTRIBUTED)

  85.     message("Imperative configured to link megray")

  86.     target_link_libraries(${MODULE_NAME} PRIVATE megray)

  87. endif()

  88. target_include_directories(${MODULE_NAME} PUBLIC src/include PRIVATE ${PYTHON_INCLUDE_DIRS} ${NUMPY_INCLUDE_DIR})

  89. target_compile_definitions(${MODULE_NAME} PRIVATE MODULE_NAME=${MODULE_NAME})

  90. target_compile_options(${MODULE_NAME} PRIVATE -Wno-unused-parameter)

  91. if(CXX_SUPPORT_WCLASS_MEMACCESS)

  92.     target_compile_options(${MODULE_NAME} PRIVATE "-Wno-class-memaccess")

  93. endif()

  94. set_target_properties(${MODULE_NAME} PROPERTIES

  95.     SUFFIX ${CMAKE_SHARED_LIBRARY_SUFFIX}

  96.     LIBRARY_OUTPUT_DIRECTORY ${MEGENGINE_DIR}/${PACKAGE_NAME}/core

  97. )

  98. if (APPLE OR MSVC OR WIN32)

  99.     message("-- overwriting SUFFIX at macos and windows before config by set_target_properties")

  100.     pybind11_extension(${MODULE_NAME})

  101. endif()

  102. add_dependencies(${MODULE_NAME} gen_opr_py _version_ld)

  103. 

  104. if(MGE_WITH_TEST AND MGE_ENABLE_RTTI)

  105.     add_subdirectory(test)

  106. endif()

  107. 

  108. add_custom_command(

  109.     TARGET ${MODULE_NAME} POST_BUILD

  110.     COMMAND ${CMAKE_COMMAND} -E copy ${PROJECT_SOURCE_DIR}/LICENSE ${PROJECT_SOURCE_DIR}/ACKNOWLEDGMENTS ${PROJECT_BINARY_DIR}

  111.     COMMAND ${CMAKE_COMMAND} -E copy_directory ${CMAKE_CURRENT_SOURCE_DIR}/python/megengine ${CMAKE_CURRENT_BINARY_DIR}/python/megengine

  112.     COMMAND ${CMAKE_COMMAND} -E copy_directory ${CMAKE_CURRENT_SOURCE_DIR}/python/test ${CMAKE_CURRENT_BINARY_DIR}/python/test

  113.     COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_CURRENT_SOURCE_DIR}/python/setup.py ${CMAKE_CURRENT_BINARY_DIR}/python/setup.py

  114.     COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_CURRENT_SOURCE_DIR}/python/requires.txt ${CMAKE_CURRENT_BINARY_DIR}/python/requires.txt

  115.     COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_CURRENT_SOURCE_DIR}/python/requires-style.txt ${CMAKE_CURRENT_BINARY_DIR}/python/requires-style.txt

  116.     COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_CURRENT_SOURCE_DIR}/python/requires-test.txt ${CMAKE_CURRENT_BINARY_DIR}/python/requires-test.txt

  117. )