| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 71 std::vector<size_t> output_size_list; in TbeOpParallelBuild() local
72 (void)TbeKernelBuild::GetIOSize(kernel_json, &input_size_list, &output_size_list); in TbeOpParallelBuild()
75 if (build_manger->SearchInCache(json_name, input_size_list, output_size_list, anf_node.get()) && in TbeOpParallelBuild()
81 build_manger->SaveSameOpInfo(anf_node, json_name, input_size_list, output_size_list); in TbeOpParallelBuild()
88 build_manger->SaveTaskInfo(task_id, anf_node, json_name, input_size_list, output_size_list); in TbeOpParallelBuild()
128 … const std::vector<size_t> &output_size_list, int64_t scope_id) { in SaveTaskInfo() argument
139 task_info.output_size_list.assign(output_size_list.begin(), output_size_list.end()); in SaveTaskInfo()
189 …= GenKernelMod(task_iter->second.input_size_list, task_iter->second.output_size_list, kernel_pack); in TaskFinishProcess()
220 const std::vector<size_t> &output_size_list) { in SaveSameOpInfo() argument
226 task_info.output_size_list.assign(output_size_list.begin(), output_size_list.end()); in SaveSameOpInfo()
[all …]
|
| D | tbe_kernel_parallel_build.h | 38 std::vector<size_t> output_size_list; member
47 … const std::vector<size_t> &input_size_list, const std::vector<size_t> &output_size_list,
51 … const std::vector<size_t> &input_size_list, const std::vector<size_t> &output_size_list);
53 … const std::vector<size_t> &input_size_list, const std::vector<size_t> &output_size_list);
57 const std::vector<size_t> &output_size_list, AnfNode *node) const;
62 …nKernelMod(const std::vector<size_t> &input_size_list, const std::vector<size_t> &output_size_list,
|
| D | tbe_kernel_build.cc | 849 void GetOutputSizeList(const nlohmann::json &output_json, std::vector<size_t> *output_size_list) { in GetOutputSizeList() argument
850 MS_EXCEPTION_IF_NULL(output_size_list); in GetOutputSizeList()
874 output_size_list->push_back(size_i); in GetOutputSizeList()
880 std::vector<size_t> *output_size_list) { in GetIOSize() argument
881 if (input_size_list == nullptr || output_size_list == nullptr) { in GetIOSize()
886 output_size_list->clear(); in GetIOSize()
888 GetOutputSizeList(kernel_json[kJOpInfo][kJOutputs], output_size_list); in GetIOSize()
931 void GetRealOutputSize(const nlohmann::json &output_json, std::vector<size_t> *output_size_list, si… in GetRealOutputSize() argument
947 output_size_list->push_back((*size_i)); in GetRealOutputSize()
950 void GetOutputSizeList2(const nlohmann::json &output_json, std::vector<size_t> *output_size_list) { in GetOutputSizeList2() argument
[all …]
|
| D | tbe_kernel_build.h | 41 std::vector<size_t> *output_size_list, const AnfNodePtr &anf_node);
43 std::vector<size_t> *output_size_list);
49 … std::vector<size_t> *input_size_list, std::vector<size_t> *output_size_list);
53 std::vector<size_t> *output_size_list);
|
| D | ascend_kernel_compile.cc | 672 std::vector<size_t> output_size_list; in AscendFusionOpCompile() local
674 &output_size_list)) { in AscendFusionOpCompile()
681 … auto kernel_mod = build_manager_->GenKernelMod(input_size_list, output_size_list, kernel_pack); in AscendFusionOpCompile()
692 input_size_list, output_size_list); in AscendFusionOpCompile()
709 build_manager_->SaveTaskInfo(task_id, nullptr, json_name, input_size_list, output_size_list, in AscendFusionOpCompile()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 89 std::vector<size_t> output_size_list; in KernelFusion() local
91 &output_size_list)) { in KernelFusion()
97 auto kernel_mod = build_manger->GenKernelMod(input_size_list, output_size_list, kernel_pack); in KernelFusion()
106 output_size_list); in KernelFusion()
124 build_manger->SaveTaskInfo(task_id, nullptr, json_name, input_size_list, output_size_list, in KernelFusion()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | kernel_actor.cc | 348 const auto &output_size_list = kernel_mod->GetOutputSizeList(); in FetchOutputDeviceTensor() local
350 if (output_addresses.size() != output_size_list.size()) { in FetchOutputDeviceTensor()
357 if (output_size_list[i] != output_address->GetSize()) { in FetchOutputDeviceTensor()
359 output_address->SetSize(output_size_list[i]); in FetchOutputDeviceTensor()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_build.cc | 42 kernel_mod_ptr->SetOutputSizeList(json_generator.output_size_list()); in AkgSetKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_build.cc | 44 kernel_mod_ptr->SetOutputSizeList(json_generator.output_size_list()); in AkgSetKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_build.cc | 81 std::vector<size_t> output_size_list; in SetIOSize() local
99 output_size_list.push_back(LongToSize(size_i)); in SetIOSize()
101 kernel_mod_ptr->SetOutputSizeList(output_size_list); in SetIOSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.h | 106 const std::vector<size_t> &output_size_list() const { return output_size_list_; } in output_size_list() function
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 381 const auto &output_size_list = kernel_mode->GetOutputSizeList(); in FetchMemUnitSize() local
385 …size_t output_size = std::accumulate(output_size_list.begin(), output_size_list.end(), kZeroNumber… in FetchMemUnitSize()
|