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 {
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
#endif // MINDSPORE_CCSRC_IR_PRIMITIVE_H_

Loading…
Cancel
Save