Browse Source

fix static-check problems

pull/15919/head
liuyihong 5 years ago
parent
commit
68ffa6789b
2 changed files with 7 additions and 7 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc
  2. +6
    -6
      mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc View File

@@ -88,7 +88,7 @@ class OpInfoExtractor {
}
}

bool ExcludeAttr(const std::string &name) {
bool ExcludeAttr(const std::string &name) const {
const std::set<std::string> black_list = {"IsFeatureMapInputList", "IsFeatureMapOutput", kAttrOutputNames,
kAttrInputNames};
return black_list.count(name) != 0;


+ 6
- 6
mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h View File

@@ -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<size_t, size_t> &position);
TypeId GetInputDataType(const AnfNodePtr &anf_node, size_t real_index);
std::vector<size_t> 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<size_t> 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<size_t> 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<size_t> 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);


Loading…
Cancel
Save