Browse Source

!593 [ir] fix prim hash function

Merge pull request !593 from biffex/fix-prim-hash-function
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
a468dc09a6
1 changed files with 1 additions and 4 deletions
  1. +1
    -4
      mindspore/ccsrc/ir/primitive.h

+ 1
- 4
mindspore/ccsrc/ir/primitive.h View File

@@ -152,10 +152,7 @@ struct PrimitiveEqual {
}; };


struct PrimitiveHasher { struct PrimitiveHasher {
std::size_t operator()(PrimitivePtr const &prim) const {
std::size_t hash = std::hash<std::string>()(prim->name());
return hash;
}
std::size_t operator()(PrimitivePtr const &prim) const { return prim->Hash(); }
}; };
} // namespace mindspore } // namespace mindspore
#endif // MINDSPORE_CCSRC_IR_PRIMITIVE_H_ #endif // MINDSPORE_CCSRC_IR_PRIMITIVE_H_

Loading…
Cancel
Save