| @@ -364,7 +364,6 @@ bool TbeKernelSelect::GenBuilderItem(bool is_input, size_t kernel_build_info_ind | |||||
| } | } | ||||
| dynamic_input_index++; | dynamic_input_index++; | ||||
| real_io_tensor_index += dynamic_input_size; | real_io_tensor_index += dynamic_input_size; | ||||
| } else { | } else { | ||||
| if (ios_info.size() != 1) { | if (ios_info.size() != 1) { | ||||
| MS_LOG(EXCEPTION) << "if output is dynamic, so output must has one output."; | MS_LOG(EXCEPTION) << "if output is dynamic, so output must has one output."; | ||||
| @@ -18,6 +18,7 @@ | |||||
| #include <map> | #include <map> | ||||
| #include <memory> | #include <memory> | ||||
| #include <string> | #include <string> | ||||
| #include <algorithm> | |||||
| #include "utility" | #include "utility" | ||||
| #include "backend/session/anf_runtime_algorithm.h" | #include "backend/session/anf_runtime_algorithm.h" | ||||
| #include "runtime/mem.h" | #include "runtime/mem.h" | ||||
| @@ -226,7 +227,7 @@ void DataDumper::ConstructDumpTask(NotNull<const CNodePtr &> kernel, NotNull<aic | |||||
| DumpKernelInput(kernel, args, dump_task); | DumpKernelInput(kernel, args, dump_task); | ||||
| } | } | ||||
| void DataDumper::SetOpDebugMappingInfo(NotNull<aicpu::dump::OpMappingInfo *> dump_info) const { | |||||
| void DataDumper::SetOpDebugMappingInfo(const NotNull<aicpu::dump::OpMappingInfo *> dump_info) const { | |||||
| MS_LOG(INFO) << "[DataDump] Add op debug info to OpMappingInfo, task id = " << debug_task_id_ | MS_LOG(INFO) << "[DataDump] Add op debug info to OpMappingInfo, task id = " << debug_task_id_ | ||||
| << ", stream id = " << debug_stream_id_; | << ", stream id = " << debug_stream_id_; | ||||
| aicpu::dump::Task task; | aicpu::dump::Task task; | ||||
| @@ -61,7 +61,7 @@ class DataDumper { | |||||
| void ReleaseDevMem(void **ptr) const; | void ReleaseDevMem(void **ptr) const; | ||||
| bool KernelNeedDump(const CNodePtr &kernel) const; | bool KernelNeedDump(const CNodePtr &kernel) const; | ||||
| void SetOpMappingInfo(NotNull<aicpu::dump::OpMappingInfo *> dump_info) const; | void SetOpMappingInfo(NotNull<aicpu::dump::OpMappingInfo *> dump_info) const; | ||||
| void SetOpDebugMappingInfo(NotNull<aicpu::dump::OpMappingInfo *> dump_info) const; | |||||
| void SetOpDebugMappingInfo(const NotNull<aicpu::dump::OpMappingInfo *> dump_info) const; | |||||
| void ConstructDumpTask(NotNull<const CNodePtr &> kernel, NotNull<aicpu::dump::Task *> dump_task) const; | void ConstructDumpTask(NotNull<const CNodePtr &> kernel, NotNull<aicpu::dump::Task *> dump_task) const; | ||||
| std::function<void *()> model_handle_; | std::function<void *()> model_handle_; | ||||
| @@ -1705,6 +1705,5 @@ void DfGraphConvertor::RegisterAdapter(const std::string &name, OpAdapterPtr adp | |||||
| void DfGraphConvertor::RegisterAdapter(const std::string &name, OpAdapterPtr train_adpt, OpAdapterPtr infer_adpt) { | void DfGraphConvertor::RegisterAdapter(const std::string &name, OpAdapterPtr train_adpt, OpAdapterPtr infer_adpt) { | ||||
| OpAdapterMap::get()[name] = std::make_shared<OpAdapterDesc>(train_adpt, infer_adpt); | OpAdapterMap::get()[name] = std::make_shared<OpAdapterDesc>(train_adpt, infer_adpt); | ||||
| } | } | ||||
| } // namespace transform | } // namespace transform | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -202,7 +202,6 @@ class OpAdapterMap { | |||||
| private: | private: | ||||
| static std::unordered_map<std::string, OpAdapterDescPtr> adpt_map_; | static std::unordered_map<std::string, OpAdapterDescPtr> adpt_map_; | ||||
| }; | }; | ||||
| } // namespace transform | } // namespace transform | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| #endif // MINDSPORE_CCSRC_TRANSFORM_GRAPH_IR_OP_ADAPTER_MAP_H_ | #endif // MINDSPORE_CCSRC_TRANSFORM_GRAPH_IR_OP_ADAPTER_MAP_H_ | ||||