| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_all_to_all.cc | 95 void *workspace_addr = nullptr; in GenTask() local
101 workspace_addr = workspace.at(0)->addr; in GenTask()
106 workspace_addr, task.workspace_size, task.stream_num, private_def, in GenTask()
|
| D | hccl_kernel.cc | 256 void *workspace_addr = nullptr; in GenTask() local
262 workspace_addr = workspace.at(0)->addr; in GenTask()
267 … output_data_addr, workspace_addr, task.workspace_size, task.stream_num, in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | multinomial_gpu_kernel.h | 54 void *workspace_addr = GetDeviceAddress<void *>(workspace, 1); in Launch() local
56 curandState *devStates = reinterpret_cast<curandState *>(workspace_addr); in Launch()
|
| D | random_op_gpu_kernel.h | 71 void *workspace_addr = GetDeviceAddress<void *>(workspace, 0); in Launch() local
72 curandState *devStates = reinterpret_cast<curandState *>(workspace_addr); in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | array_reduce_gpu_kernel.h | 52 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
67 …cudnnReduceTensor(cudnn_handle_, reduce_tensor_descriptor_, nullptr, 0, workspace_addr, workspace_… in Launch()
75 …cudnnReduceTensor(cudnn_handle_, reduce_tensor_descriptor_, nullptr, 0, workspace_addr, workspace_… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | instance_norm_gpu_kernel.h | 78 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 4); in Launch() local
93 workspace_addr, workspace_size_, reserve_addr, 0), in Launch()
|
| D | bias_add_grad_gpu_kenel.h | 66 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 1); in Launch() local
71 … cudnnReduceTensor(cudnn_handle_, op_desc_, indices_addr, workspace_size_list_[0], workspace_addr, in Launch()
|
| D | conv3d_gpu_kernel.h | 48 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
60 … conv_algorithm_, workspace_addr, workspace_size_, &beta, output_desc_, output_addr), in Launch()
66 … conv_algorithm_, workspace_addr, workspace_size_, &beta, output_desc_, output_addr), in Launch()
|
| D | conv2d_gpu_kernel.h | 49 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
67 … conv_algorithm_, workspace_addr, workspace_size_, &beta, output_desc_, output_addr), in Launch()
73 … conv_algorithm_, workspace_addr, workspace_size_, &beta, output_desc_, output_addr), in Launch()
|
| D | l2normalize_gpu_kernel.h | 63 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 1); in Launch() local
76 …cudnnReduceTensor(cudnn_handle_, reduce_tensor_descriptor_, nullptr, 0, workspace_addr, workspace_… in Launch()
|
| D | instance_norm_grad_gpu_kernel.h | 79 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 3); in Launch() local
97 … save_variance, activation_desc_, workspace_addr, workspace_size_, reserve_addr, 0), in Launch()
|
| D | lstm_grad_weight_gpu_kernel.h | 68 void *workspace_addr = GetDeviceAddress<T>(workspace, 0); in Launch() local
85 … y_addr, workspace_addr, workspace_size_list_[0], dw_desc_, dw_addr, reserved_addr, in Launch()
|
| D | batch_norm_gpu_kernel.h | 59 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
72 workspace_addr, workspace_size_, reserve_addr, reserve_size_), in Launch()
|
| D | lstm_gpu_kernel.h | 77 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
90 … workspace_addr, workspace_size_list_[0], reserved_addr, reserved_size_), in Launch()
|
| D | l2normalize_grad_gpu_kernel.h | 67 T *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 2); in Launch() local
81 …cudnnReduceTensor(cudnn_handle_, reduce_tensor_descriptor_, nullptr, 0, workspace_addr, workspace_… in Launch()
|
| D | batch_norm_grad_gpu_kernel.h | 96 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
106 … epsilon_, save_mean, save_variance, activation_desc_, workspace_addr, in Launch()
|
| D | lstm_grad_data_gpu_kernel.h | 81 void *workspace_addr = GetPossiblyNullDeviceAddress<T>(workspace, 0); in Launch() local
95 … dx_desc_.get(), dx_addr, dhx_desc_, dhx_addr, dcx_desc_, dcx_addr, workspace_addr, in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 232 … void *output_data_addr, void *workspace_addr, int64_t workspace_size, int64_t hccl_stream_num, in HcclTaskInfo() argument
239 workspace_addr_(workspace_addr), in HcclTaskInfo()
254 void *workspace_addr() const { return workspace_addr_; } in workspace_addr() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | rolling_cpu_kernel.cc | 161 auto workspace_addr = reinterpret_cast<size_t *>(workspace[0]->addr); in Launch() local
179 size_t *ids = workspace_addr + offset; in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | hccl_task.cc | 66 workspace_mem_ = task_info_->workspace_addr(); in Distribute()
|