Home
last modified time | relevance | path

Searched refs:workspace_size (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dcholesky_trsm_solve_gpu_kernel.h104 size_t workspace_size = batch_ * sizeof(T *); in InitSizeLists() local
105 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
106 workspace_size = batch_ * sizeof(T *); in InitSizeLists()
107 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
108 workspace_size = batch_ * sizeof(int); in InitSizeLists()
109 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
116 size_t workspace_size = batch_ * sizeof(T *); in InitSizeLists() local
117 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
118 workspace_size = batch_ * sizeof(T *); in InitSizeLists()
119 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
[all …]
Dcholesky_solve_gpu_kernel.h239 size_t workspace_size; in InitSizeLists() local
244 workspace_size = batch_ * m_ * lda_ * unit_size; in InitSizeLists()
245 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
250 workspace_size = batch_ * sizeof(T *); in InitSizeLists()
251 (void)workspace_size_list_.insert(workspace_size_list_.begin(), workspace_size); in InitSizeLists()
252 workspace_size = batch_ * sizeof(T *); in InitSizeLists()
253 (void)workspace_size_list_.insert(workspace_size_list_.begin(), workspace_size); in InitSizeLists()
254 workspace_size = batch_ * sizeof(int); in InitSizeLists()
255 (void)workspace_size_list_.insert(workspace_size_list_.begin(), workspace_size); in InitSizeLists()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_delegate_fp32.h96 size_t workspace_size() override { in workspace_size() function
97 InnerKernel::workspace_size(); in workspace_size()
98 return conv_kernel_->workspace_size(); in workspace_size()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_delegate_fp16.h58 size_t workspace_size() override { in workspace_size() function
59 InnerKernel::workspace_size(); in workspace_size()
60 return fp16_conv_kernel_->workspace_size(); in workspace_size()
Dconvolution_delegate_fp16.cc121 set_workspace_size(fp16_conv_kernel_->workspace_size()); in ReSize()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhcom_all_to_all.cc86 << ", workspace_size=" << task.workspace_size << ", stream_num=" << task.stream_num in GenTask()
96 if (task.workspace_size != 0) { in GenTask()
106 workspace_addr, task.workspace_size, task.stream_num, private_def, in GenTask()
Dhccl_kernel.cc247 << ", workspace_size=" << task.workspace_size << ", stream_num=" << task.stream_num in GenTask()
257 if (task.workspace_size != 0) { in GenTask()
267 … output_data_addr, workspace_addr, task.workspace_size, task.stream_num, in GenTask()
/third_party/mindspore/mindspore/lite/src/
Dinner_kernel.cc28 workspace_ = malloc(workspace_size()); in AllocWorkspace()
30 MS_LOG(ERROR) << "fail to alloc " << workspace_size() << "in kernel" << name(); in AllocWorkspace()
Dinner_kernel.h161 virtual size_t workspace_size() { return workspace_size_; } in workspace_size() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/
Dbatchnorm_fold2_grad_gpu_kernel.h136 size_t workspace_size = batch_size_ * channel_ * sizeof(T); in InitSizeLists() local
152 workspace_size_list_.push_back(workspace_size); // tmp in InitSizeLists()
153 workspace_size_list_.push_back(workspace_size); // tmp2 in InitSizeLists()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dbias_add_grad_gpu_kenel.h200 size_t indices_size, workspace_size; in InitSizeLists() local
205 …ode_, cudnnGetReductionWorkspaceSize(cudnn_handle_, op_desc_, dy_desc_, db_desc_, &workspace_size), in InitSizeLists()
208 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
Ddropout3d_gpu_kernel.h140 size_t workspace_size = num_chan_ * sizeof(float); // rand_f for curandGen in InitSizeLists() local
141 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
Dlstm_grad_weight_gpu_kernel.h202 size_t workspace_size = 0; in InitSizeLists() local
204 … cudnnGetRNNWorkspaceSize(handle_, rnn_desc_, seq_len_, x_desc_.get(), &workspace_size), in InitSizeLists()
206 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
Dlstm_gpu_kernel.h254 size_t workspace_size = 0; in InitSizeLists() local
256 … cudnnGetRNNWorkspaceSize(handle_, rnn_desc_, seq_len_, x_desc_.get(), &workspace_size), in InitSizeLists()
258 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
Dlstm_grad_data_gpu_kernel.h252 size_t workspace_size = 0; in InitSizeLists() local
254 … cudnnGetRNNWorkspaceSize(handle_, rnn_desc_, seq_len_, dx_desc_.get(), &workspace_size), in InitSizeLists()
256 workspace_size_list_.push_back(workspace_size); in InitSizeLists()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_session.cc96 size_t workspace_size = 0; in AllocWorkSpace() local
98 if (workspace_size < static_cast<kernel::InnerKernel *>(kernel->kernel())->workspace_size()) { in AllocWorkSpace()
99 workspace_size = static_cast<kernel::InnerKernel *>(kernel->kernel())->workspace_size(); in AllocWorkSpace()
102 workspace_ = malloc(workspace_size); in AllocWorkSpace()
104 MS_LOG(ERROR) << "cannot allocate " << workspace_size << " for workspace"; in AllocWorkSpace()
/third_party/mindspore/include/c_api/
Dmodel_c.h63 …_API void OH_AI_ModelSetWorkspace(OH_AI_ModelHandle model, void *workspace, size_t workspace_size);
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/
Dtask_info.h232 … void *output_data_addr, void *workspace_addr, int64_t workspace_size, int64_t hccl_stream_num, in HcclTaskInfo() argument
240 workspace_size_(workspace_size), in HcclTaskInfo()
255 int64_t workspace_size() const { return workspace_size_; } in workspace_size() function
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/
Dhccl_task.cc65 if (task_info_->workspace_size() > 0) { in Distribute()
79 ge_task.kernelHcclInfo[0].workSpaceMemSize = task_info_->workspace_size(); in Distribute()
/third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/
Dconverter.cc223 int64_t workspace_size = workspace_sizes[0]; in ParseDomiTask() local
231 return {task_def.private_def(), workspace_size, stream_num}; in ParseDomiTask()
Dhccl_adapter.cc258 int64_t workspace_size = workspace_sizes[0]; in CalcWorkspaceSize() local
259 MS_LOG(INFO) << "Node " << node->DebugString() << " workspace size is " << workspace_size; in CalcWorkspaceSize()
261 return workspace_size; in CalcWorkspaceSize()
Dhccl_adapter.h38 int64_t workspace_size; member
/third_party/glib/glib/pcre/
Dpcre_compile.c689 int newsize = cd->workspace_size * 2; in expand_workspace()
692 if (cd->workspace_size >= COMPILE_WORK_SIZE_MAX || in expand_workspace()
693 newsize - cd->workspace_size < WORK_SIZE_SAFETY_MARGIN) in expand_workspace()
698 memcpy(newspace, cd->start_workspace, cd->workspace_size * sizeof(pcre_uchar)); in expand_workspace()
700 if (cd->workspace_size > COMPILE_WORK_SIZE) in expand_workspace()
703 cd->workspace_size = newsize; in expand_workspace()
3559 if (code > cd->start_workspace + cd->workspace_size - in compile_branch()
3610 else if (cd->hwm > cd->start_workspace + cd->workspace_size - in compile_branch()
5254 while (cd->hwm > cd->start_workspace + cd->workspace_size - in compile_branch()
5336 while (cd->hwm > cd->start_workspace + cd->workspace_size - in compile_branch()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc386 …size_t workspace_size = std::accumulate(workspace_size_list.begin(), workspace_size_list.end(), kZ… in FetchMemUnitSize() local
387 current_sum_size = current_sum_size + input_size + output_size + workspace_size; in FetchMemUnitSize()
415 auto output_workspace_size = output_size + workspace_size; in FetchMemUnitSize()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dconvolution_base.cc166 memset(packed_weight_, 0, workspace_size()); in RepackWeight()

12