Searched refs:kernel_mod_ (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_info.cc | 50 if (kernel_mod_ == nullptr && index >= output_address_list_.size()) { in SetOutputAddr()
54 } else if (kernel_mod_ != nullptr && output_address_list_.empty()) { in SetOutputAddr()
56 for (size_t i = 0; i < kernel_mod_->GetOutputSizeList().size(); i++) { in SetOutputAddr()
94 if (kernel_mod_ == nullptr) { in SetWorkspaceAddr()
98 for (size_t i = 0; i < kernel_mod_->GetWorkspaceSizeList().size(); i++) { in SetWorkspaceAddr()
111 void KernelInfo::set_kernel_mod(const kernel::KernelModPtr &kernel_mod) { kernel_mod_ = kernel_mod;… in set_kernel_mod()
113 kernel::KernelMod *KernelInfo::MutableKernelMod() const { return kernel_mod_.get(); } in MutableKernelMod()
115 const kernel::KernelMod *KernelInfo::kernel_mod() const { return kernel_mod_.get(); } in kernel_mod()
132 if ((kernel_mod_ == nullptr && other.kernel_mod_ != nullptr) || in operator ==()
133 (kernel_mod_ != nullptr && other.kernel_mod_ == nullptr)) { in operator ==()
|
| D | launch_kernel.cc | 78 MS_EXCEPTION_IF_NULL(kernel_mod_); in LaunchSingleKernel()
80 auto kernel_inputs = ObtainKernelInputs(kernel_mod_->GetInputSizeList(), inputs_addr); in LaunchSingleKernel()
82 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchSingleKernel()
84 auto kernel_workspaces = ObtainKernelWorkspaces(kernel_mod_->GetWorkspaceSizeList()); in LaunchSingleKernel()
86 auto ret_status = kernel_mod_->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in LaunchSingleKernel()
|
| D | launch_kernel.h | 27 explicit LaunchKernel(void *stream) : stream_(stream), kernel_mod_(nullptr) {} in LaunchKernel()
46 kernel::KernelMod *kernel_mod_; variable
|
| D | kernel_info.h | 34 kernel_mod_ = nullptr; in KernelInfo()
81 kernel::KernelModPtr kernel_mod_; variable
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_atomic_clean.cc | 51 kernel_mod_ = AnfAlgo::GetKernelMod(atomic_clean_graph_->execution_order()[0]); in LaunchOpKernel()
52 MS_EXCEPTION_IF_NULL(kernel_mod_); in LaunchOpKernel()
62 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchOpKernel()
64 auto kernel_workspaces = ObtainKernelWorkspaces(kernel_mod_->GetWorkspaceSizeList()); in LaunchOpKernel()
66 auto ret_status = kernel_mod_->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in LaunchOpKernel()
|
| D | ascend_launch_transdata.cc | 53 kernel_mod_ = AnfAlgo::GetKernelMod(transdata_graph_->execution_order()[0]); in LaunchOpKernel()
54 MS_EXCEPTION_IF_NULL(kernel_mod_); in LaunchOpKernel()
64 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchOpKernel()
66 auto kernel_workspace = ObtainKernelWorkspaces(kernel_mod_->GetWorkspaceSizeList()); in LaunchOpKernel()
68 auto ret_status = kernel_mod_->Launch(kernel_inputs, kernel_workspace, kernel_outputs, stream_); in LaunchOpKernel()
|
| D | ascend_launch_mul.cc | 41 kernel_mod_ = ObtainLaunchMulKernelMod(); in LaunchOpKernel()
42 MS_EXCEPTION_IF_NULL(kernel_mod_); in LaunchOpKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_launch_mul.cc | 40 kernel_mod_ = ObtainLaunchMulKernelMod(); in LaunchOpKernel()
41 MS_EXCEPTION_IF_NULL(kernel_mod_); in LaunchOpKernel()
|