| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | converter.cc | 219 auto workspace_sizes = op->GetWorkspaceBytes(); in ParseDomiTask() local
220 if (workspace_sizes.size() != 1) { in ParseDomiTask()
221 MS_LOG(EXCEPTION) << "Unexpected workspace size " << workspace_sizes.size(); in ParseDomiTask()
223 int64_t workspace_size = workspace_sizes[0]; in ParseDomiTask()
|
| D | hccl_adapter.cc | 254 auto workspace_sizes = op->GetWorkspaceBytes(); in CalcWorkspaceSize() local
255 if (workspace_sizes.size() != 1) { in CalcWorkspaceSize()
256 …MS_LOG(EXCEPTION) << "Unexpected workspace size " << workspace_sizes.size() << ", which should be … in CalcWorkspaceSize()
258 int64_t workspace_size = workspace_sizes[0]; in CalcWorkspaceSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | apply_momentum_ps_kernel.cc | 31 const std::vector<size_t> &ApplyMomentumPSKernel::workspace_sizes() const { return GetWorkspaceSize… in workspace_sizes() function in mindspore::kernel::ps::ApplyMomentumPSKernel
|
| D | apply_momentum_ps_kernel.h | 39 const std::vector<size_t> &workspace_sizes() const override;
|
| D | sparse_apply_adam_ps_kernel.h | 43 const std::vector<size_t> &workspace_sizes() const override;
|
| D | sparse_apply_lazy_adam_ps_kernel.h | 42 const std::vector<size_t> &workspace_sizes() const override;
|
| D | embedding_look_up_ps_kernel.h | 43 const std::vector<size_t> &workspace_sizes() const override;
|
| D | sparse_apply_ftrl_ps_kernel.h | 44 const std::vector<size_t> &workspace_sizes() const override;
|
| D | pserver_kernel.h | 45 virtual const std::vector<size_t> &workspace_sizes() const = 0;
|
| D | embedding_look_up_ps_kernel.cc | 111 const std::vector<size_t> &EmbeddingLookUpPSKernel::workspace_sizes() const { return GetWorkspaceSi… in workspace_sizes() function in mindspore::kernel::ps::EmbeddingLookUpPSKernel
|
| D | sparse_apply_adam_ps_kernel.cc | 115 const std::vector<size_t> &SparseApplyAdamPSKernel::workspace_sizes() const { return GetWorkspaceSi… in workspace_sizes() function in mindspore::kernel::ps::SparseApplyAdamPSKernel
|
| D | sparse_apply_lazy_adam_ps_kernel.cc | 117 const std::vector<size_t> &SparseApplyLazyAdamPSKernel::workspace_sizes() const { return GetWorkspa… in workspace_sizes() function in mindspore::kernel::ps::SparseApplyLazyAdamPSKernel
|
| D | sparse_apply_ftrl_ps_kernel.cc | 126 const std::vector<size_t> &SparseApplyFtrlPSKernel::workspace_sizes() const { return GetWorkspaceSi… in workspace_sizes() function in mindspore::kernel::ps::SparseApplyFtrlPSKernel
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 640 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in InitKernelWorkspaceAddress() local
641 for (size_t i = 0; i < workspace_sizes.size(); ++i) { in InitKernelWorkspaceAddress()
642 auto device_address = CreateDeviceAddress(nullptr, workspace_sizes[i], "", kTypeUnknown); in InitKernelWorkspaceAddress()
705 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in ClearKernelWorkspaceAddress() local
706 for (size_t i = 0; i < workspace_sizes.size(); ++i) { in ClearKernelWorkspaceAddress()
1165 auto workspace_sizes = kernel_mod.GetWorkspaceSizeList(); in AllocKernelWorkspaceDynamicRes() local
1166 for (size_t i = 0; i < workspace_sizes.size(); ++i) { in AllocKernelWorkspaceDynamicRes()
1167 if (workspace_sizes[i] == 0) { in AllocKernelWorkspaceDynamicRes()
1173 …if (device_address->ptr_ == nullptr && !AttemptMallocMem(device_address, workspace_sizes[i], mock)… in AllocKernelWorkspaceDynamicRes()
1179 workspace->size = workspace_sizes[i]; in AllocKernelWorkspaceDynamicRes()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 206 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in CreateKernelWorkspaceDeviceAddress() local
207 for (size_t i = 0; i < workspace_sizes.size(); ++i) { in CreateKernelWorkspaceDeviceAddress()
208 …auto device_address = device_context->CreateDeviceAddress(nullptr, workspace_sizes[i], "", kTypeUn… in CreateKernelWorkspaceDeviceAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 168 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in AssignKernelOutputAddress() local
169 for (size_t i = 0; i < workspace_sizes.size(); ++i) { in AssignKernelOutputAddress()
170 …AnfAlgo::SetWorkspaceAddr(CreateDeviceAddress(nullptr, workspace_sizes[i], kOpFormat_DEFAULT, kNum… in AssignKernelOutputAddress()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse.cc | 111 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in InitDynamicWorkspaceKernelRef() local
113 for (auto size : workspace_sizes) { in InitDynamicWorkspaceKernelRef()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.cc | 36 std::vector<size_t> ws_sizes = pserver_kernel->workspace_sizes(); in Build()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 307 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in ResetNodeAddress() local
308 for (size_t i = 0; i < workspace_sizes.size(); ++i) { in ResetNodeAddress()
309 …AnfAlgo::SetWorkspaceAddr(CreateDeviceAddress(nullptr, workspace_sizes[i], kOpFormat_DEFAULT, kNum… in ResetNodeAddress()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 466 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in InitSomasOutputAndWorkspaceTensors() local
468 for (const auto &size : workspace_sizes) { in InitSomasOutputAndWorkspaceTensors()
|