Searched refs:is_task_sink (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.h | 55 bool Load(const session::KernelGraph &graph, bool is_task_sink) override;
56 bool Run(const session::KernelGraph &graph, bool is_task_sink) override;
|
| D | ascend_kernel_runtime.cc | 413 bool AscendKernelRuntime::Load(const session::KernelGraph &graph, bool is_task_sink) { in Load() argument
414 if (!is_task_sink) { in Load()
701 bool AscendKernelRuntime::Run(const session::KernelGraph &graph, bool is_task_sink) { in Run() argument
706 if (is_task_sink) { in Run()
1081 bool is_task_sink = context_ptr->get_param<bool>(MS_CTX_ENABLE_TASK_SINK); in HcclInit() local
1083 if (!is_task_sink && mode == kGraphMode) { in HcclInit()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | comm_manager.cc | 76 bool is_task_sink = context_ptr->get_param<bool>(MS_CTX_ENABLE_TASK_SINK); in CreateGroupSync() local
78 if (!is_task_sink && mode == kGraphMode) { in CreateGroupSync()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.h | 68 virtual bool Load(const session::KernelGraph &graph, bool is_task_sink);
69 virtual bool Run(const session::KernelGraph &graph, bool is_task_sink) = 0;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel.cc | 198 bool is_task_sink = context_ptr->get_param<bool>(MS_CTX_ENABLE_TASK_SINK); in GetWorkspaceSizeList() local
200 …if (!workspace_size_list_.empty() || hccl_data_type_list_.empty() || (!is_task_sink && mode == kGr… in GetWorkspaceSizeList()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.h | 39 bool Run(const session::KernelGraph &graph, bool is_task_sink) override;
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.h | 47 bool Run(const session::KernelGraph &graph, bool is_task_sink) override;
|
| D | gpu_kernel_runtime.cc | 456 bool GPUKernelRuntime::Run(const session::KernelGraph &graph, bool is_task_sink) { in Run() argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_session.cc | 1513 bool is_task_sink = context_ptr->get_param<bool>(MS_CTX_ENABLE_TASK_SINK); in Load() local
1517 bool ret_ok = runtime_instance->Load(*kernel_graph, is_task_sink); in Load()
1526 bool is_task_sink = false; in Execute() local
1530 is_task_sink = context_ptr->get_param<bool>(MS_CTX_ENABLE_TASK_SINK); in Execute()
1534 if (is_task && is_task_sink) { in Execute()
1539 bool ret_ok = runtime_instance->Run(*kernel_graph, is_task_sink); in Execute()
1540 if (is_task && is_task_sink) { in Execute()
|