[NODE][IR] Introduce StructuralHash for the Unified IR.

include/tvm/ir/adt.h

@@ -71,6 +71,11 @@ class ConstructorNode : public RelayExprNode {
         equal(inputs, other->inputs);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(name_hint);
+    hash_reduce(inputs);
+  }
+
   static constexpr const char* _type_key = "relay.Constructor";
   TVM_DECLARE_FINAL_OBJECT_INFO(ConstructorNode, RelayExprNode);
 };
@@ -123,6 +128,12 @@ class TypeDataNode : public TypeNode {
         equal(constructors, other->constructors);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce.DefHash(header);
+    hash_reduce.DefHash(type_vars);
+    hash_reduce(constructors);
+  }
+
   static constexpr const char* _type_key = "relay.TypeData";
   TVM_DECLARE_FINAL_OBJECT_INFO(TypeDataNode, TypeNode);
 };

include/tvm/ir/attrs.h

@@ -121,6 +121,7 @@ class AttrFieldInfoNode : public Object {
 
   static constexpr const char* _type_key = "AttrFieldInfo";
   static constexpr bool _type_has_method_sequal_reduce = false;
+  static constexpr bool _type_has_method_shash_reduce = false;
   TVM_DECLARE_FINAL_OBJECT_INFO(AttrFieldInfoNode, Object);
 };
 
@@ -281,6 +282,7 @@ class BaseAttrsNode : public Object {
   TVM_DLL virtual size_t ContentHash(AttrsHash hasher) const = 0;
 
   static constexpr const bool _type_has_method_sequal_reduce = true;
+  static constexpr const bool _type_has_method_shash_reduce = true;
   static constexpr const char* _type_key = "Attrs";
   TVM_DECLARE_BASE_OBJECT_INFO(BaseAttrsNode, Object);
 };
@@ -309,6 +311,10 @@ class DictAttrsNode : public BaseAttrsNode {
     return equal(dict, other->dict);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(dict);
+  }
+
   // implementations
   void VisitAttrs(AttrVisitor* v) final;
   void VisitNonDefaultAttrs(AttrVisitor* v) final;
@@ -452,6 +458,21 @@ class AttrsHashVisitor {
   const AttrsHash& hasher_;
 };
 
+class AttrsSHashVisitor {
+ public:
+  explicit AttrsSHashVisitor(const SHashReducer& hash_reducer)
+      : hash_reducer_(hash_reducer) {}
+
+  template<typename T>
+  AttrNopEntry operator()(const char* key, T* value) {
+    hash_reducer_(*value);
+    return AttrNopEntry();
+  }
+
+ private:
+  const SHashReducer& hash_reducer_;
+};
+
 // helper entry that does initialization, set default.
 template<typename T>
 struct AttrInitEntry {
@@ -858,6 +879,11 @@ class AttrsNode : public BaseAttrsNode {
     return visitor.result_;
   }
 
+  void SHashReduce(SHashReducer hash_reducer) const {
+    ::tvm::detail::AttrsSHashVisitor visitor(hash_reducer);
+    self()->__VisitAttrs__(visitor);
+  }
+
   Array<AttrFieldInfo> ListFieldInfo() const final {
     ::tvm::detail::AttrDocVisitor visitor;
     self()->__VisitAttrs__(visitor);

include/tvm/ir/env_func.h

@@ -52,11 +52,18 @@ class EnvFuncNode : public Object {
   }
 
   bool SEqualReduce(const EnvFuncNode* other, SEqualReducer equal) const {
-    return this == other;
+    // name uniquely identifies the env function.
+    return name == other->name;
+  }
+
+  void SHashReduce(SHashReducer hash_reduce) const {
+    // Name uniquely identifies the env function.
+    hash_reduce(name);
   }
 
   static constexpr const char* _type_key = "EnvFunc";
   static constexpr bool _type_has_method_sequal_reduce = true;
+  static constexpr bool _type_has_method_shash_reduce = true;
   TVM_DECLARE_FINAL_OBJECT_INFO(EnvFuncNode, Object);
 };
 

include/tvm/ir/expr.h

@@ -44,6 +44,7 @@ class BaseExprNode : public Object {
  public:
   static constexpr const char* _type_key = "Expr";
   static constexpr const bool _type_has_method_sequal_reduce = true;
+  static constexpr const bool _type_has_method_shash_reduce = true;
   TVM_DECLARE_BASE_OBJECT_INFO(BaseExprNode, Object);
 };
 
@@ -205,6 +206,11 @@ class GlobalVarNode : public RelayExprNode {
         equal.FreeVarEqualImpl(this, other);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(name_hint);
+    hash_reduce.FreeVarHashImpl(this);
+  }
+
   static constexpr const char* _type_key = "GlobalVar";
   TVM_DECLARE_FINAL_OBJECT_INFO(GlobalVarNode, RelayExprNode);
 };
@@ -240,6 +246,11 @@ class IntImmNode : public PrimExprNode {
     return equal(dtype, other->dtype) && equal(value, other->value);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(dtype);
+    hash_reduce(value);
+  }
+
   static constexpr const char* _type_key = "IntImm";
   TVM_DECLARE_FINAL_OBJECT_INFO(IntImmNode, PrimExprNode);
 };
@@ -279,6 +290,11 @@ class FloatImmNode : public PrimExprNode {
     return equal(dtype, other->dtype) && equal(value, other->value);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(dtype);
+    hash_reduce(value);
+  }
+
   static constexpr const char* _type_key = "FloatImm";
   TVM_DECLARE_FINAL_OBJECT_INFO(FloatImmNode, PrimExprNode);
 };
@@ -373,8 +389,14 @@ class RangeNode : public Object {
     return equal(min, other->min) && equal(extent, other->extent);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(min);
+    hash_reduce(extent);
+  }
+
   static constexpr const char* _type_key = "Range";
   static constexpr const bool _type_has_method_sequal_reduce = true;
+  static constexpr const bool _type_has_method_shash_reduce = true;
   TVM_DECLARE_FINAL_OBJECT_INFO(RangeNode, Object);
 };
 

include/tvm/ir/module.h

@@ -64,6 +64,8 @@ class IRModuleNode : public Object {
 
   TVM_DLL bool SEqualReduce(const IRModuleNode* other, SEqualReducer equal) const;
 
+  TVM_DLL void SHashReduce(SHashReducer hash_reduce) const;
+
   /*!
    * \brief Add a function to the global environment.
    * \param var The var of the global function.
@@ -238,6 +240,7 @@ class IRModuleNode : public Object {
 
   static constexpr const char* _type_key = "IRModule";
   static constexpr const bool _type_has_method_sequal_reduce = true;
+  static constexpr const bool _type_has_method_shash_reduce = true;
   TVM_DECLARE_FINAL_OBJECT_INFO(IRModuleNode, Object);
 
  private:

include/tvm/ir/op.h

@@ -106,6 +106,11 @@ class OpNode : public RelayExprNode {
     return this == other;
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    // Name uniquely identifies an Op.
+    hash_reduce(name);
+  }
+
   /*!
    * \brief Check that if current op is a "primtive operator".
    * That is the arguments are all type variables, and there is a single

include/tvm/ir/tensor_type.h

@@ -79,6 +79,11 @@ class TensorTypeNode : public BaseTensorTypeNode {
         equal(dtype, other->dtype);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(shape);
+    hash_reduce(dtype);
+  }
+
   /*! \brief Return product of elements in the shape.
    *  \return (d1 * d_2 ... * d_n) if shape is (d_1, d_2, ..., d_n) and 1 if shape size is zero.
    */

include/tvm/ir/type.h

@@ -80,6 +80,7 @@ class TypeNode : public Object {
 
   static constexpr const char* _type_key = "Type";
   static constexpr const bool _type_has_method_sequal_reduce = true;
+  static constexpr const bool _type_has_method_shash_reduce = true;
   TVM_DECLARE_BASE_OBJECT_INFO(TypeNode, Object);
 };
 
@@ -115,6 +116,10 @@ class PrimTypeNode : public TypeNode {
     return equal(dtype, other->dtype);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(dtype);
+  }
+
   static constexpr const char* _type_key = "PrimType";
   TVM_DECLARE_FINAL_OBJECT_INFO(PrimTypeNode, TypeNode);
 };
@@ -161,6 +166,10 @@ class PointerTypeNode : public TypeNode {
     return equal(element_type, other->element_type);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(element_type);
+  }
+
   static constexpr const char* _type_key = "PointerType";
   TVM_DECLARE_FINAL_OBJECT_INFO(PointerTypeNode, TypeNode);
 };
@@ -233,6 +242,11 @@ class TypeVarNode : public TypeNode {
         equal.FreeVarEqualImpl(this, other);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(kind);
+    hash_reduce.FreeVarHashImpl(this);
+  }
+
   static constexpr const char* _type_key = "TypeVar";
   TVM_DECLARE_FINAL_OBJECT_INFO(TypeVarNode, TypeNode);
 };
@@ -280,6 +294,11 @@ class GlobalTypeVarNode : public TypeNode {
         equal.FreeVarEqualImpl(this, other);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(name_hint);
+    hash_reduce.FreeVarHashImpl(this);
+  }
+
   static constexpr const char* _type_key = "GlobalTypeVar";
   TVM_DECLARE_FINAL_OBJECT_INFO(GlobalTypeVarNode, TypeNode);
 };
@@ -320,6 +339,10 @@ class TupleTypeNode : public TypeNode {
     return equal(fields, other->fields);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(fields);
+  }
+
   static constexpr const char* _type_key = "TupleType";
   TVM_DECLARE_FINAL_OBJECT_INFO(TupleTypeNode, TypeNode);
 };
@@ -421,6 +444,13 @@ class FuncTypeNode : public TypeNode {
         equal(type_constraints, other->type_constraints);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce.DefHash(type_params);
+    hash_reduce(arg_types);
+    hash_reduce(ret_type);
+    hash_reduce(type_constraints);
+  }
+
   static constexpr const char* _type_key = "FuncType";
   TVM_DECLARE_FINAL_OBJECT_INFO(FuncTypeNode, TypeNode);
 };
@@ -471,6 +501,10 @@ class IncompleteTypeNode : public TypeNode {
     return equal(kind, other->kind);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(kind);
+  }
+
   static constexpr const char* _type_key = "IncompleteType";
   TVM_DECLARE_FINAL_OBJECT_INFO(IncompleteTypeNode, TypeNode);
 };
@@ -512,6 +546,10 @@ class RelayRefTypeNode : public TypeNode {
     return equal(value, other->value);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(value);
+  }
+
   // Keep the relay prefix in the type as this type is specific
   // to the relay itself.
   static constexpr const char* _type_key = "relay.RefType";

include/tvm/ir/type_relation.h

@@ -56,6 +56,11 @@ class TypeCallNode : public TypeNode {
         equal(args, other->args);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(func);
+    hash_reduce(args);
+  }
+
   static constexpr const char* _type_key = "TypeCall";
   TVM_DECLARE_FINAL_OBJECT_INFO(TypeCallNode, TypeNode);
 };
@@ -209,6 +214,13 @@ class TypeRelationNode : public TypeConstraintNode {
         equal(attrs, other->attrs);
   }
 
+  void SHashReduce(SHashReducer hash_reduce) const {
+    hash_reduce(func);
+    hash_reduce(args);
+    hash_reduce(num_inputs);
+    hash_reduce(attrs);
+  }
+
   static constexpr const char* _type_key = "TypeRelation";
   TVM_DECLARE_FINAL_OBJECT_INFO(TypeRelationNode, TypeConstraintNode);
 };

include/tvm/node/node.h

@@ -41,6 +41,7 @@
 #include <tvm/node/repr_printer.h>
 #include <tvm/node/container.h>
 #include <tvm/node/structural_equal.h>
+#include <tvm/node/structural_hash.h>
 
 #include <string>
 #include <vector>