| @@ -23,7 +23,7 @@ option(ENABLE_PYTHON "Enable python" ON) | |||||
| option(ENABLE_ACL "enable acl" OFF) | option(ENABLE_ACL "enable acl" OFF) | ||||
| option(ENABLE_GLIBCXX "enable_glibcxx" OFF) | option(ENABLE_GLIBCXX "enable_glibcxx" OFF) | ||||
| if(NOT ENABLE_D AND NOT ENABLE_TESTCASES AND NOT ENABLE_ACL) | |||||
| if(NOT ENABLE_D AND NOT ENABLE_TESTCASES AND NOT ENABLE_ACL AND NOT ENABLE_GE) | |||||
| set(ENABLE_GLIBCXX ON) | set(ENABLE_GLIBCXX ON) | ||||
| endif() | endif() | ||||
| @@ -256,7 +256,7 @@ if(ENABLE_GE) | |||||
| else() | else() | ||||
| target_link_libraries(mindspore ge_client) | target_link_libraries(mindspore ge_client) | ||||
| endif() | endif() | ||||
| target_link_libraries(mindspore graph tsdclient datatransfer) | |||||
| target_link_libraries(mindspore graph tsdclient datatransfer ascendcl runtime) | |||||
| endif() | endif() | ||||
| if(ENABLE_D) | if(ENABLE_D) | ||||
| @@ -80,7 +80,7 @@ using mindspore::abstract::AbstractTensorPtr; | |||||
| using mindspore::abstract::AbstractTuple; | using mindspore::abstract::AbstractTuple; | ||||
| using mindspore::abstract::AbstractTuplePtr; | using mindspore::abstract::AbstractTuplePtr; | ||||
| #if (ENABLE_GE || ENABLE_D) | |||||
| #if (ENABLE_D) | |||||
| using mindspore::device::ascend::ProfilingManager; | using mindspore::device::ascend::ProfilingManager; | ||||
| #endif | #endif | ||||
| @@ -1109,7 +1109,7 @@ void InitHccl() { | |||||
| (void)context::OpenTsd(ms_context); | (void)context::OpenTsd(ms_context); | ||||
| } | } | ||||
| #endif | #endif | ||||
| #if (ENABLE_GE || ENABLE_D) | |||||
| #if (ENABLE_D) | |||||
| if (!ProfilingManager::GetInstance().IsProfiling()) { | if (!ProfilingManager::GetInstance().IsProfiling()) { | ||||
| ProfilingManager::GetInstance().SetHcclEnabledBefProfilingEnabled(); | ProfilingManager::GetInstance().SetHcclEnabledBefProfilingEnabled(); | ||||
| } | } | ||||
| @@ -16,8 +16,11 @@ else() | |||||
| endif() | endif() | ||||
| if(ENABLE_D) | if(ENABLE_D) | ||||
| file(GLOB_RECURSE D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "ascend/*.cc" "kernel_adjust.cc" | |||||
| "../../minddata/dataset/engine/tdt/tdt_handle.cc") | |||||
| file(GLOB_RECURSE D_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "ascend/*.cc" "kernel_adjust.cc") | |||||
| endif() | |||||
| if(ENABLE_TDTQUE) | |||||
| file(GLOB_RECURSE TDT_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | |||||
| "../../minddata/dataset/engine/tdt/tdt_handle.cc") | |||||
| endif() | endif() | ||||
| if(ENABLE_CPU) | if(ENABLE_CPU) | ||||
| @@ -75,4 +78,4 @@ endif() | |||||
| list(REMOVE_ITEM D_SRC_LIST "ascend/profiling/profiling_callback_register.cc") | list(REMOVE_ITEM D_SRC_LIST "ascend/profiling/profiling_callback_register.cc") | ||||
| set_property(SOURCE ${DEVICE_SRC_LIST} ${D_SRC_LIST} ${CPU_SRC_LIST} | set_property(SOURCE ${DEVICE_SRC_LIST} ${D_SRC_LIST} ${CPU_SRC_LIST} | ||||
| PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_DEVICE) | PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_DEVICE) | ||||
| add_library(_mindspore_runtime_device_obj OBJECT ${DEVICE_SRC_LIST} ${D_SRC_LIST} ${CPU_SRC_LIST}) | |||||
| add_library(_mindspore_runtime_device_obj OBJECT ${DEVICE_SRC_LIST} ${D_SRC_LIST} ${CPU_SRC_LIST} ${TDT_SRC_LIST}) | |||||