Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 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: d90e121547
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'd90e121547'
${ noResults }
mindspore/mindspore/ccsrc/CMakeLists.txt

499 lines
19 kB
Raw Blame History 

  1. include_directories(${CMAKE_CURRENT_SOURCE_DIR})

  2. if(ENABLE_CPU)

  3.     include(ExternalProject)

  4.     add_compile_definitions(CPUSESSION)

  5.     file(GLOB_RECURSE CPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  6.     "device/cpu/*.cc"

  7.     )

  8. endif()

  9. 

  10. if(ENABLE_GPU)

  11.     find_package(CUDA REQUIRED)

  12.     find_package(Threads)

  13.     if(${CUDA_VERSION} VERSION_LESS ${MS_REQUIRE_CUDA_VERSION})

  14.       message(FATAL_ERROR "The minimum CUDA version ${MS_REQUIRE_CUDA_VERSION} is required, but only CUDA ${CUDA_VERSION} found.")

  15.     endif()

  16.     enable_language(CUDA)

  17.     if(NOT CUDA_PATH OR CUDA_PATH STREQUAL "")

  18.         if(DEFINED ENV{CUDA_HOME})

  19.             set(CUDA_PATH $ENV{CUDA_HOME})                

  20.         else()

  21.             set(CUDA_PATH ${CUDA_TOOLKIT_ROOT_DIR})

  22.         endif()

  23.     endif()

  24.     if(NOT CUDNN_PATH OR CUDNN_PATH STREQUAL "")

  25.         set(CUDNN_PATH ${CUDA_PATH})

  26.     endif()

  27.     message("CUDA_PATH: ${CUDA_PATH}")

  28.     message("CUDA_INCLUDE_DIRS: ${CUDA_INCLUDE_DIRS}")

  29.     message("CUDNN_PATH: ${CUDNN_PATH}")

  30.     include_directories(${CUDNN_PATH} ${CUDA_PATH} ${CUDA_INCLUDE_DIRS})

  31. 

  32.     file(GLOB_RECURSE GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  33.             "device/gpu/*.cc"

  34.             "device/gpu/*.cu"

  35.             "kernel/gpu/*.cu"

  36.             "kernel/akg/gpu/*.cc"

  37.             "kernel/akg/akgkernelbuild.cc"

  38.             "kernel/akg/akg_kernel_attrs_process.cc"

  39.             )

  40.     file(GLOB_RECURSE GPU_KERNEL_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  41.             "kernel/gpu/*.cc"

  42.             )

  43.     list(APPEND CUDA_NVCC_FLAGS -arch=sm_53)

  44.     list(REMOVE_ITEM GPU_SRC_LIST "device/gpu/blocking_queue.cc" "device/gpu/gpu_buffer_mgr.cc")

  45.     add_library(gpu_queue SHARED "device/gpu/blocking_queue.cc" "device/gpu/gpu_buffer_mgr.cc")

  46.     target_link_libraries(gpu_queue ${CMAKE_THREAD_LIBS_INIT} ${CUDA_PATH}/lib64/libcudart.so)

  47. 

  48. 

  49.     file(GLOB_RECURSE MS_STEPS_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  50.         "session/gpu_session.cc"

  51.         )

  52.     list(REMOVE_ITEM GPU_SRC_LIST "device/gpu/mpi/mpi_initializer.cc"

  53.                                   "device/gpu/distribution/collective_wrapper.cc"

  54.                                   "device/gpu/distribution/mpi_wrapper.cc"

  55.                                   "device/gpu/distribution/nccl_wrapper.cc"

  56.                                   )

  57.     list(REMOVE_ITEM GPU_KERNEL_SRC_LIST "device/gpu/mpi/mpi_initializer.cc"

  58.             "kernel/gpu/nccl/nccl_gpu_kernel.cc"

  59.             )

  60. 

  61.     set(NVCC_TMP_CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS})

  62.     string(REPLACE "-std=c++17" "-std=c++11" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")

  63.     cuda_add_library(gpu_cuda_lib STATIC ${GPU_SRC_LIST})

  64.     set(CMAKE_CXX_FLAGS ${NVCC_TMP_CMAKE_CXX_FLAGS})

  65. 

  66.     if(ENABLE_MPI)

  67.       include(ExternalProject)

  68. 

  69.       file(GLOB_RECURSE GPU_NCCL_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  70.               "kernel/gpu/nccl/*.cc"

  71.               )

  72.       file(GLOB_RECURSE GPU_MPI_PYTHON_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  73.               "device/gpu/mpi/mpi_initializer.cc"

  74.               )

  75.       add_library(gpu_collective SHARED "device/gpu/distribution/collective_wrapper.cc"

  76.                                         "device/gpu/distribution/mpi_wrapper.cc"

  77.                                         "device/gpu/distribution/nccl_wrapper.cc"

  78.                                         )

  79.     endif()

  80. endif()

  81. 

  82. include_directories("${CMAKE_BINARY_DIR}/predict/schema/inner")

  83. file(GLOB_RECURSE FLATBUFFER_IN RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "predict/schema/*.fbs")

  84. set(FLATBUFFER_OU "${CMAKE_BINARY_DIR}/predict/schema/inner")

  85. ms_build_flatbuffers("${FLATBUFFER_IN}" "${FLATBUFFER_IN}" GENERATED_OUTPUT_DIR "${FLATBUFFER_OU}")

  86. 

  87. file(GLOB_RECURSE MINDSPORE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  88.         "ir/*.cc"

  89.         "ir/dtype/*.cc"

  90.         "utils/context/ms_context.cc"

  91.         "utils/symbolic.cc"

  92.         "utils/tensorprint_utils.cc"

  93.         "utils/convert_utils.cc"

  94.         "utils/graph_utils.cc"

  95.         "utils/misc.cc"

  96.         "utils/callbacks.cc"

  97.         "utils/profile.cc"

  98.         "utils/base_ref.cc"

  99.         "utils/summary/event_writer.cc"

  100.         "utils/log_adapter.cc"

  101.         "utils/comm_manager.cc"

  102.         "utils/any.cc"

  103.         "utils/config_manager.cc"

  104.         "utils/system/file_system.cc"

  105.         "utils/system/crc32c.cc"

  106.         "common/*.cc"

  107.         "parallel/*.cc"

  108.         "pipeline/pipeline.cc"

  109.         "pipeline/resource.cc"

  110.         "pipeline/pass.cc"

  111.         "pipeline/action.cc"

  112.         "pipeline/validator.cc"

  113.         "pipeline/remove_value_node_dup.cc"

  114.         "pipeline/parse/*.cc"

  115.         "pipeline/static_analysis/*.cc"

  116.         "optimizer/*.cc"

  117.         "debug/*.cc"

  118.         "onnx/onnx_exporter.cc"

  119.         "operator/*.cc"

  120.         "session/kernel_graph.cc"

  121.         "utils/node_utils.cc"

  122.         "session/session_basic.cc"

  123.         "session/session_factory.cc"

  124.         "session/anf_runtime_algorithm.cc"

  125.         "vm/*.cc"

  126.         "pynative/base.cc"

  127.         "pynative/pynative_execute.cc"

  128.         "pybind_api/*.cc"

  129.         "device/common/*.cc"

  130.         "kernel/kernel_query.cc"

  131.         "kernel/kernel_build_info.cc"

  132.         "kernel/kash/*.cc"

  133.         "device/kernel_info.cc"

  134.         "device/kernel_runtime.cc"

  135.         "device/memory_manager.cc"

  136.         "device/kernel_runtime_manager.cc"

  137.         "device/convert_tensor_utils.cc"

  138.         "pre_activate/common/*.cc"

  139.         "pre_activate/pass/*.cc"

  140.         "pre_activate/gpu/*.cc"

  141.         "pre_activate/mem_reuse/*.cc"

  142.         "predict/predict.cc"

  143.         "predict/generator/utils/ir_model_util.cc"

  144.         "predict/converter/*.cc"

  145.         "predict/converter/attr_utils/*.cc"

  146.         "predict/converter/lite_model/*.cc"

  147.         "predict/converter/lite_model/operations/*.cc"

  148.         "kernel/common_utils.cc"

  149.         "kernel/oplib/*.cc"

  150.         "kernel/kash/*.cc"

  151.         "device/gpu/distribution/collective_init.cc"

  152.         )

  153. 

  154. file(GLOB_RECURSE MEM_REUSE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  155.         "pre_activate/mem_reuse/*.cc"

  156.         )

  157. if(NOT ENABLE_DUMP_E2E)

  158.     list(REMOVE_ITEM MINDSPORE_SRC_LIST "debug/e2e_dump.cc")

  159. endif()

  160. file(COPY "${ms_onnx_INC}/onnx/onnx.proto" DESTINATION ${CMAKE_CURRENT_SOURCE_DIR})

  161. file(GLOB_RECURSE ONNX_PROTO RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "${CMAKE_CURRENT_SOURCE_DIR}/onnx.proto")

  162. message("onnx proto path is : ${ONNX_PROTO}")

  163. ms_protobuf_generate(ONNX_PROTO_SRCS ONNX_PROTO_HDRS ${ONNX_PROTO})

  164. list(APPEND MINDSPORE_PROTO_LIST ${ONNX_PROTO_SRCS})

  165. 

  166. if(ENABLE_DUMP_PROTO)

  167.     include_directories(${CMAKE_BINARY_DIR})

  168.     file(GLOB_RECURSE PROTO_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  169.             "utils/node_strategy.proto"

  170.             )

  171.     ms_protobuf_generate(PROTO_SRCS PROTO_HDRS ${PROTO_LIST})

  172. 

  173.     file(GLOB_RECURSE PROTO_PY RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  174.             "utils/anf_ir.proto"

  175.             "utils/summary.proto"

  176.             "utils/checkpoint.proto"

  177.             )

  178.     ms_protobuf_generate_py(PY_SRCS PY_HDRS PY_PYS ${PROTO_PY})

  179. 

  180.     list(APPEND MINDSPORE_PROTO_DUMP_LIST ${PROTO_SRCS})

  181.     list(APPEND MINDSPORE_PROTO_DUMP_LIST ${PY_SRCS})

  182.     list(APPEND MINDSPORE_SRC_LIST "debug/dump_proto.cc")

  183.     list(APPEND MINDSPORE_SRC_LIST "parallel/strategy_checkpoint/parallel_strategy_checkpoint.cc")

  184.     add_compile_definitions(ENABLE_DUMP_PROTO)

  185. endif()

  186. 

  187. if(ENABLE_GE)

  188.     file(GLOB_RECURSE GE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  189.             "transform/*.cc"

  190.             "pynative/pynative_execute_ge.cc"

  191.             "utils/callbacks_ge.cc"

  192.             "pipeline/pipeline_ge.cc"

  193.             )

  194.     list(APPEND MINDSPORE_SRC_LIST ${GE_SRC_LIST})

  195. endif()

  196. 

  197. if(ENABLE_D)

  198.     include_directories("${CMAKE_BINARY_DIR}/kernel/aicpu")

  199.     file(GLOB_RECURSE PROTO_IN RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  200.             "kernel/aicpu/proto/*.proto"

  201.             )

  202.     ms_protobuf_generate(PROTOSRCS PROTOHDRS ${PROTO_IN})

  203. 

  204.     include_directories("${CMAKE_BINARY_DIR}/predict/generator/ir")

  205.     file(GLOB_RECURSE PROTO_INNER RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  206.             "predict/proto/*.proto"

  207.             )

  208.     ms_protobuf_generate(PREDICT_PROTOSRCS PREDICT_PROTOHDRS ${PROTO_INNER})

  209. 

  210.     file(GLOB_RECURSE D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  211.             "device/ascend/*.cc"

  212.             "device/ascend/profiling/*.cc"

  213.             "device/ascend/tasksink/*.cc"

  214.             "device/kernel_adjust.cc"

  215.             "kernel/kernel_fusion.cc"

  216.             "kernel/tbe/*.cc"

  217.             "pre_activate/ascend/*.cc"

  218.             "transform/*.cc"

  219.             "pipeline/pipeline_ge.cc"

  220.             )

  221.     list(APPEND MINDSPORE_SRC_LIST ${D_SRC_LIST})

  222.     list(APPEND MINDSPORE_PROTO_AICPU_LIST ${PROTOSRCS})

  223.     list(APPEND MINDSPORE_PROTO_PREDICT_LIST ${PREDICT_PROTOSRCS})

  224. 

  225.     file(GLOB_RECURSE MS_STEPS_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  226.         "session/ascend_session.cc"

  227.         )

  228.     file(GLOB_RECURSE MS_TASKINFO_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  229.         "device/ascend/tasksink/taskinfo/*.cc")

  230.     file(GLOB_RECURSE MS_AICPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  231.         "kernel/aicpu/*.cc"

  232.         )

  233.     file(GLOB_RECURSE MS_RT_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  234.         "kernel/mng/*.cc"

  235.         )

  236.     file(GLOB_RECURSE MS_HCCL_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  237.         "kernel/hccl/*.cc"

  238.         )

  239.     file(GLOB_RECURSE MS_PREDICT_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  240.         "predict/generator/ir/*.cc"

  241.         )

  242.     add_compile_definitions(ENABLE_D)

  243. endif()

  244. 

  245. file(GLOB_RECURSE MS_GVAR_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  246.         "gvar/*.cc"

  247.         )

  248. 

  249. add_library(mindspore_gvar SHARED ${MS_GVAR_SRC_LIST})

  250. add_library(mindspore STATIC ${MINDSPORE_SRC_LIST})

  251. 

  252. if(ENABLE_D)

  253.     list(APPEND MINDSPORE_PROTO_LIST ${MINDSPORE_PROTO_AICPU_LIST})

  254. endif()

  255. if(ENABLE_DUMP_PROTO)

  256.     list(APPEND MINDSPORE_PROTO_LIST ${MINDSPORE_PROTO_DUMP_LIST})

  257. endif()

  258. list(APPEND MINDSPORE_PROTO_LIST ${MINDSPORE_PROTO_PREDICT_LIST})

  259. if(MINDSPORE_PROTO_LIST)

  260.     add_library(proto_input STATIC ${MINDSPORE_PROTO_LIST})

  261.     set_target_properties(proto_input PROPERTIES COMPILE_FLAGS "-Wno-unused-variable")

  262.     target_link_libraries(mindspore proto_input)

  263. endif()

  264. 

  265. if(APPLE)

  266.     set_target_properties(mindspore_gvar PROPERTIES MACOSX_RPATH ON)

  267. endif()

  268. 

  269. link_directories(${CMAKE_SOURCE_DIR}/build/mindspore/graphengine)

  270. 

  271. if (ENABLE_GE)

  272.     if(ENABLE_TRAIN)

  273.         target_link_libraries(mindspore graph ge_client_train)

  274.     else()

  275.         target_link_libraries(mindspore graph ge_client)

  276.     endif()

  277.     target_link_libraries(mindspore tsdclient)

  278. elseif(ENABLE_D)

  279.     add_compile_definitions(NO_GE_CLIENT)

  280.     target_link_libraries(mindspore graph)

  281. else()

  282.     add_compile_definitions(NO_GE_CLIENT)

  283. endif()

  284. 

  285. if(ENABLE_D)

  286.     if (DEFINED ENV{D_LINK_PATH})

  287.         if (CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "aarch64")

  288.             MESSAGE("system processor matches aarch64")

  289.             set(D_LIB_PATH $ENV{D_LINK_PATH}/aarch64)

  290.         elseif (CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "x86_64")

  291.             MESSAGE("system processor matches x86_64")

  292.             set(D_LIB_PATH $ENV{D_LINK_PATH}/x86_64)

  293.         else ()

  294.             MESSAGE("system ${CMAKE_HOST_SYSTEM_PROCESSOR} not support")

  295.         endif()

  296.     else()

  297.         MESSAGE("use system default lib")

  298.         if(DEFINED ENV{ASCEND_CUSTOM_PATH})

  299.             set(ASCEND_PATH $ENV{ASCEND_CUSTOM_PATH})

  300.         else()

  301.             set(ASCEND_PATH /usr/local/Ascend)

  302.         endif()

  303.         set(ASCEND_DRIVER_PATH ${ASCEND_PATH}/driver/lib64/common)

  304.         set(ASCEND_DRIVER_BACK_PATH ${ASCEND_PATH}/driver/lib64/driver)

  305.         set(ASCEND_RUNTIME_PATH ${ASCEND_PATH}/fwkacllib/lib64)

  306.     endif()

  307. 

  308.     MESSAGE("USE DAV LIB PATH: ${ASCEND_PATH}")

  309.     find_library(HCCL hccl ${ASCEND_RUNTIME_PATH})

  310.     find_library(CCE_LIB cce ${ASCEND_RUNTIME_PATH})

  311.     find_library(RUNTIME_LIB runtime ${ASCEND_RUNTIME_PATH})

  312.     find_library(TSDCLIENT tsdclient HINTS ${ASCEND_RUNTIME_PATH} ${ASCEND_DRIVER_BACK_PATH})

  313.     find_library(PROFILING msprof ${ASCEND_DRIVER_PATH})

  314.     target_link_libraries(mindspore ge_runtime ${CCE_LIB} ${RUNTIME_LIB} ${TSDCLIENT} ${PROFILING} ${HCCL} ${TSDCLIENT})

  315. endif()

  316. 

  317. target_link_libraries(mindspore securec)

  318. target_link_libraries(mindspore dl)

  319. target_link_libraries(mindspore mindspore::flatbuffers)

  320. # link protobuf

  321. if (ENABLE_D)

  322.     target_link_libraries(mindspore mindspore::protobuf)

  323. endif()

  324. 

  325. # set c_expression building

  326. set(PYTHON_MODULE_SOURCE

  327.         pipeline/init.cc

  328.         kernel/oplib/oplib.cc

  329.     ${MS_STEPS_SRC_LIST} ${MS_CCE_SRC_LIST} ${MS_AICPU_SRC_LIST} ${MS_TASKINFO_LIST} ${MS_RT_SRC_LIST}

  330.     ${GPU_NCCL_LIST} ${MS_HCCL_SRC_LIST} ${MS_PREDICT_SRC_LIST} ${CPU_SRC_LIST} ${MEM_REUSE_SRC_LIST} ${GPU_KERNEL_SRC_LIST})

  331. 

  332. set(CMAKE_BUILD_WITH_INSTALL_RPATH TRUE)

  333. pybind11_add_module(_c_expression ${PYTHON_MODULE_SOURCE})

  334. 

  335. MESSAGE(STATUS "operation system is ${CMAKE_SYSTEM}")

  336. if (CMAKE_SYSTEM_NAME MATCHES "Linux")

  337.     target_link_options(_c_expression PRIVATE -Wl,-init,mindspore_log_init)

  338.     set(ORIGIN_PATH $ORIGIN)

  339. elseif (CMAKE_SYSTEM_NAME MATCHES "Darwin")

  340.     set_target_properties(_c_expression PROPERTIES MACOSX_RPATH ON)

  341.     set(ORIGIN_PATH @loader_path)

  342. else ()

  343.     MESSAGE(FATAL_ERROR "other platform: ${CMAKE_SYSTEM_NAME}")

  344. endif ()

  345. 

  346. 

  347. set(ORIGIN_PATH ${ORIGIN_PATH}/lib)

  348. set_target_properties(_c_expression PROPERTIES INSTALL_RPATH ${ORIGIN_PATH})

  349. target_link_libraries(_c_expression PRIVATE

  350.         mindspore::pybind11_module

  351.         mindspore

  352.         mindspore_gvar

  353.         )

  354. 

  355. if(ENABLE_GPU)

  356. 	execute_process(COMMAND bash ${CMAKE_SOURCE_DIR}/third_party/apply_patches.sh 

  357. 		${CMAKE_BINARY_DIR} 

  358. 		${dlpack_DIRPATH} 

  359. 		${dmlc_core_DIRPATH} 

  360. 		${rang_DIRPATH} 

  361. 		${incubator_tvm_gpu_DIRPATH})

  362.     set(TVM_DIR "${CMAKE_BINARY_DIR}/incubator-tvm")

  363.     # Utility functions

  364.     include(${TVM_DIR}/cmake/util/Util.cmake)

  365.     include(${TVM_DIR}/cmake/util/FindCUDA.cmake)

  366. 

  367.     # include directories

  368.     include_directories(AFTER "${TVM_DIR}/include")

  369.     include_directories(AFTER "${TVM_DIR}/src")

  370.     include_directories(AFTER "${TVM_DIR}")

  371.     include_directories(AFTER "${TVM_DIR}/src/schedule")

  372. 

  373.     include_directories(AFTER "${TVM_DIR}/3rdparty/dmlc-core/include")

  374.     include_directories(AFTER "${TVM_DIR}/3rdparty/dlpack/include")

  375.     include_directories(AFTER "${TVM_DIR}/3rdparty/compiler-rt")

  376.     include_directories(AFTER "${TVM_DIR}/3rdparty/rang/include")

  377. 

  378.     # lib contain dlopen and dlclose

  379.     set(TVM_RUNTIME_LINKER_LIBS ${CMAKE_DL_LIBS})

  380. 

  381.     # add source group

  382.     file(GLOB_RECURSE GROUP_SOURCE "${TVM_DIR}/src/*.cc" "src/*.cc")

  383.     file(GLOB_RECURSE GROUP_INCLUDE "${TVM_DIR}/src/*.h"

  384.         "${TVM_DIR}/include/*.h" "src/*.h" "include/*.h")

  385.     assign_source_group("Source" ${GROUP_SOURCE})

  386.     assign_source_group("Include" ${GROUP_INCLUDE})

  387. 

  388.     file(GLOB COMPILER_SRCS

  389.         "pre_activate/gpu/*.cc"

  390.         ${TVM_DIR}/src/api/*.cc

  391.         ${TVM_DIR}/src/arithmetic/*.cc

  392.         ${TVM_DIR}/src/autotvm/*.cc

  393.         ${TVM_DIR}/src/codegen/*.cc

  394.         ${TVM_DIR}/src/lang/*.cc

  395.         ${TVM_DIR}/src/pass/*.cc

  396.         ${TVM_DIR}/src/op/*.cc

  397.         ${TVM_DIR}/src/node/*.cc

  398.         ${TVM_DIR}/src/schedule/*.cc

  399.         ${TVM_DIR}/src/runtime/*.cc

  400.         ${TVM_DIR}/src/runtime/vm/*.cc

  401.         ${TVM_DIR}/src/runtime/vm/profiler/*.cc

  402.         ${TVM_DIR}/src/codegen/stackvm/*.cc)

  403. 

  404.     file(GLOB_RECURSE RELAY_SRCS ${TVM_DIR}/src/relay/*.cc)

  405.     list(APPEND COMPILER_SRCS ${RELAY_SRCS})

  406. 

  407.     file(GLOB DATATYPE_SRCS ${TVM_DIR}/src/codegen/datatype/*.cc)

  408.     list(APPEND COMPILER_SRCS ${DATATYPE_SRCS})

  409. 

  410.     file(GLOB COMPILER_VERILOG_SRCS ${TVM_DIR}/src/codegen/verilog/*.cc)

  411.     list(APPEND COMPILER_SRCS ${COMPILER_VERILOG_SRCS})

  412. 

  413.     file(GLOB TOPI_SRCS ${TVM_DIR}/topi/src/*.cc)

  414. 

  415.     file(GLOB RUNTIME_SRCS

  416.         ${TVM_DIR}/src/runtime/*.cc

  417.         ${TVM_DIR}/src/runtime/vm/*.cc

  418.         ${TVM_DIR}/src/runtime/stub/*.cc

  419.         ${TVM_DIR}/src/runtime/stackvm/*.cc)

  420. 

  421. 

  422.     file(GLOB COMPILER_OFF_SRCS

  423.         ${TVM_DIR}/src/codegen/opt/build_*_off.cc)

  424.     set(USE_CUDA "OFF")

  425.     if(ENABLE_GPU)

  426.         list(REMOVE_ITEM COMPILER_OFF_SRCS

  427.             ${TVM_DIR}/src/codegen/opt/build_cuda_off.cc)

  428.         set(USE_CUDA "ON")

  429.     endif()

  430.     list(APPEND COMPILER_SRCS ${COMPILER_OFF_SRCS})

  431.     # Module rules

  432.     include(${TVM_DIR}/cmake/modules/CUDA.cmake)

  433. 

  434.     set(CMAKE_C_FLAGS_AKG  -pipe -Wall -fPIC -fstack-protector-all)

  435.     set(CMAKE_C_FLAGS_AKG ${CMAKE_C_FLAGS_AKG} -Wl,-z,relro,-z,now,-z,noexecstack)

  436. 

  437.     set(CMAKE_CXX_FLAGS_AKG  -std=c++11 -pipe -Wall -fPIC -fstack-protector-all)

  438.     set(CMAKE_CXX_FLAGS_AKG ${CMAKE_CXX_FLAGS_AKG} -Wl,-z,relro,-z,now,-z,noexecstack)

  439. 

  440.     if("${CMAKE_BUILD_TYPE}" STREQUAL "Debug")

  441.         message("-- Build in Debug mode")

  442.         set(CMAKE_C_FLAGS_AKG ${CMAKE_C_FLAGS_AKG} -O0 -g -rdynamic)

  443.         set(CMAKE_CXX_FLAGS_AKG ${CMAKE_CXX_FLAGS_AKG} -O0 -g -rdynamic)

  444.     else()

  445.         message("-- Build in Release mode")

  446.         set(CMAKE_C_FLAGS_AKG ${CMAKE_C_FLAGS_AKG} -O2 -Werror)

  447.         set(CMAKE_CXX_FLAGS_AKG ${CMAKE_CXX_FLAGS_AKG} -O2 -Werror)

  448.     endif()

  449.     if(CMAKE_CXX_COMPILER_ID MATCHES "GNU" AND CMAKE_CXX_COMPILER_VERSION

  450.                                                VERSION_GREATER 7.0)

  451.         set(CMAKE_CXX_FLAGS_AKG ${CMAKE_CXX_FLAGS_AKG} -faligned-new)

  452.     endif()

  453. 

  454.     add_library(akg OBJECT ${COMPILER_SRCS} ${RUNTIME_SRCS} ${TOPI_SRCS})

  455. 

  456.     target_link_libraries(akg ${TVM_LINKER_LIBS} ${TVM_RUNTIME_LINKER_LIBS})

  457.     target_compile_options(akg PRIVATE

  458.         $<$<COMPILE_LANGUAGE:C>:${CMAKE_C_FLAGS_AKG}>

  459.         $<$<COMPILE_LANGUAGE:CXX>:${CMAKE_CXX_FLAGS_AKG}>)

  460.     target_include_directories(akg PRIVATE "${TVM_DIR}/topi/include")

  461. 

  462.     add_dependencies(_c_expression akg)

  463.     target_link_libraries(_c_expression PRIVATE akg)

  464. endif()

  465. 

  466. if(ENABLE_DUMP_PROTO)

  467.     target_link_libraries(_c_expression PRIVATE mindspore::protobuf)

  468. endif()

  469. 

  470. if(ENABLE_GPU)

  471.     message("add gpu lib to c_expression")

  472.     target_link_libraries(_c_expression PRIVATE

  473.                           gpu_cuda_lib

  474.                           gpu_queue

  475.                           cublas

  476.                           ${CUDA_PATH}/lib64/libcurand.so

  477.                           ${CUDNN_PATH}/lib64/libcudnn.so

  478.                           ${CUDA_PATH}/lib64/libcudart.so

  479.                           ${CUDA_PATH}/lib64/stubs/libcuda.so)

  480.     if(ENABLE_MPI)

  481.         pybind11_add_module(_ms_mpi ${GPU_MPI_PYTHON_LIST})

  482.         target_link_libraries(_ms_mpi PRIVATE mindspore::pybind11_module mindspore::ompi)

  483.         target_link_libraries(gpu_collective PRIVATE mindspore::ompi mindspore::nccl)

  484.     endif()

  485. endif()

  486. 

  487. 

  488. if(ENABLE_CPU)

  489.     target_link_libraries(_c_expression PRIVATE mindspore::dnnl mindspore::mkldnn)

  490. endif()

  491. 

  492. if(ENABLE_MINDDATA)

  493.     add_subdirectory(mindrecord)

  494.     add_subdirectory(dataset)

  495. endif()

  496. 

  497. if (USE_GLOG)

  498.     target_link_libraries(_c_expression PRIVATE mindspore::glog)

  499. endif()