a few more fixes

apache · Mar 29, 2020 · de1ced1 · de1ced1
1 parent de405dd
commit de1ced1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/include/tvm/ir/type.h b/include/tvm/ir/type.h
@@ -498,7 +498,9 @@ class IncompleteTypeNode : public TypeNode {
   }
 
   bool SEqualReduce(const IncompleteTypeNode* other, SEqualReducer equal) const {
-    return equal(kind, other->kind);
+    return
+        equal(kind, other->kind) &&
+        equal.FreeVarEqualImpl(this, other);
   }
 
   void SHashReduce(SHashReducer hash_reduce) const {

diff --git a/tests/python/frontend/caffe2/test_graph.py b/tests/python/frontend/caffe2/test_graph.py
@@ -15,6 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Test graph equality of caffe2 models."""
+import tvm
 from tvm import relay
 from tvm.relay import transform
 from model_zoo import c2_squeezenet, relay_squeezenet

diff --git a/tests/python/relay/test_ir_structural_equal_hash.py b/tests/python/relay/test_ir_structural_equal_hash.py
@@ -107,7 +107,7 @@ def test_func_type_sequal():
     ft = relay.FuncType(tvm.runtime.convert([t1, t2]), tp1,
                          tvm.runtime.convert([tp1, tp3]),
                          tvm.runtime.convert([tr1]))
-    translate_vars = relay.FuncType(tvm.runtime.convert([t1, t2]), tp1,
+    translate_vars = relay.FuncType(tvm.runtime.convert([t1, t2]), tp2,
                          tvm.runtime.convert([tp2, tp4]),
                          tvm.runtime.convert([tr2]))
     assert ft == translate_vars