diff --git a/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc b/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc index 7508476cb5..4630fa5fb8 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc @@ -88,7 +88,7 @@ class OpInfoExtractor { } } - bool ExcludeAttr(const std::string &name) { + bool ExcludeAttr(const std::string &name) const { const std::set black_list = {"IsFeatureMapInputList", "IsFeatureMapOutput", kAttrOutputNames, kAttrInputNames}; return black_list.count(name) != 0; diff --git a/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h b/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h index 99bca58db8..94b39844d6 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h +++ b/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h @@ -118,12 +118,12 @@ class AkgKernelJsonGenerator { nlohmann::json *node_json); std::string GetTensorName(const nlohmann::json &node_json, const std::string &tag, const std::pair &position); - TypeId GetInputDataType(const AnfNodePtr &anf_node, size_t real_index); - std::vector GetInputShape(const AnfNodePtr &anf_node, size_t real_index); - std::string GetInputFormat(const AnfNodePtr &anf_node, size_t real_index); - TypeId GetOutputDataType(const AnfNodePtr &anf_node, size_t index); - std::vector GetOutputShape(const AnfNodePtr &anf_node, size_t index); - std::string GetOutputFormat(const AnfNodePtr &anf_node, size_t index); + inline TypeId GetInputDataType(const AnfNodePtr &anf_node, size_t real_index); + inline std::vector GetInputShape(const AnfNodePtr &anf_node, size_t real_index); + inline std::string GetInputFormat(const AnfNodePtr &anf_node, size_t real_index); + inline TypeId GetOutputDataType(const AnfNodePtr &anf_node, size_t index); + inline std::vector GetOutputShape(const AnfNodePtr &anf_node, size_t index); + inline std::string GetOutputFormat(const AnfNodePtr &anf_node, size_t index); void SaveNodeAddress(const AnfNodePtr &anf_node, nlohmann::json *node_json); OpInfoPtr ExtractOpInfo(const AnfNodePtr &anf_node); void CollectParallelDimInfo(const AnfNodePtr &anf_node);