Home
last modified time | relevance | path

Searched refs:aicpu_param_head (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/
Daicpu_task.cc103 aicpu::AicpuParamHead aicpu_param_head; in SetAicpuParamHead() local
104 aicpu_param_head.length = args_size; in SetAicpuParamHead()
105 aicpu_param_head.ioAddrNum = io_addrs_num; in SetAicpuParamHead()
110 aicpu_param_head.extInfoLength = 0; in SetAicpuParamHead()
111 aicpu_param_head.extInfoAddr = 0; in SetAicpuParamHead()
125 aicpu_param_head.extInfoLength = ext_size; in SetAicpuParamHead()
126 aicpu_param_head.extInfoAddr = reinterpret_cast<uintptr_t>(ext_info_); in SetAicpuParamHead()
130 …t_ret = rtMemcpy(args_, sizeof(aicpu::AicpuParamHead), reinterpret_cast<void *>(&aicpu_param_head), in SetAicpuParamHead()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_mod.cc106 AicpuParamHead aicpu_param_head{}; in CreateCpuKernelInfo() local
107 aicpu_param_head.length = param_len; in CreateCpuKernelInfo()
108 aicpu_param_head.ioAddrNum = io_addrs_num; in CreateCpuKernelInfo()
112 aicpu_param_head.extInfoLength = 0; in CreateCpuKernelInfo()
113 aicpu_param_head.extInfoAddr = 0; in CreateCpuKernelInfo()
119 (void)args_.append(reinterpret_cast<const char *>(&aicpu_param_head), sizeof(AicpuParamHead)); in CreateCpuKernelInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dai_cpu_dynamic_kernel.cc104 auto aicpu_param_head = reinterpret_cast<kernel::AicpuParamHead *>(args_.data()); in Initialize() local
105 MS_EXCEPTION_IF_NULL(aicpu_param_head); in Initialize()
106 aicpu_param_head->extInfoLength = SizeToUint(ext_info_size_); in Initialize()
107 aicpu_param_head->extInfoAddr = reinterpret_cast<uint64_t>(ext_info_addr_dev_); in Initialize()