• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1file(GLOB_RECURSE _SESSION_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
2    "kernel_build_client.cc"
3    "kernel_graph.cc"
4    "session_basic.cc"
5    "session_factory.cc"
6    "executor.cc"
7    "executor_manager.cc"
8    "anf_runtime_algorithm.cc"
9    "single_kernel_graph.cc"
10)
11
12if("${ENABLE_HIDDEN}" STREQUAL "OFF")
13    string(REPLACE " -Werror " " " CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
14    string(REPLACE " -fvisibility=hidden" " -fvisibility=default" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
15endif()
16
17if(CMAKE_SYSTEM_NAME MATCHES "Darwin")
18    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-overloaded-virtual")
19endif()
20
21if(ENABLE_GPU)
22    file(GLOB_RECURSE _GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "gpu_session.cc" "gpu_inference_session.cc")
23    list(APPEND _SESSION_SRC_LIST ${_GPU_SRC_LIST})
24endif()
25
26if(ENABLE_CPU)
27    file(GLOB_RECURSE _CPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "cpu_session.cc")
28    list(APPEND _SESSION_SRC_LIST ${_CPU_SRC_LIST})
29endif()
30
31if(ENABLE_D)
32    file(GLOB_RECURSE _D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
33        "ascend_session.cc"
34        "ascend_control_parser.cc"
35        "ascend_auto_monad.cc"
36        "ascend_inference_session.cc"
37        )
38    list(APPEND _SESSION_SRC_LIST ${_D_SRC_LIST})
39endif()
40
41set_property(SOURCE ${_SESSION_SRC_LIST} PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_SESSION)
42add_library(_mindspore_backend_session_obj OBJECT ${_SESSION_SRC_LIST})
43