save

apache · Oct 19, 2019 · 66c9472 · 66c9472
1 parent a5cd7d2
commit 66c9472
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/include/tvm/relay/feature.h b/include/tvm/relay/feature.h
@@ -55,7 +55,7 @@ enum Feature : int {
   fLetRec = 17
 };
 
-constexpr size_t feature_count = 17;
+constexpr size_t feature_count = 18;
 
 /*!
  * \brief A finite set of Feature.

diff --git a/tests/python/relay/test_pass_partial_eval.py b/tests/python/relay/test_pass_partial_eval.py
@@ -342,12 +342,13 @@ def test_tuple_match():
 
 def test_fatal():
     msg = "user-defined fatal message"
-    assert alpha_equal(dcpe(Fatal(msg)), Fatal(msg))
+    UnitType = TupleType([])
+    assert alpha_equal(dcpe(Fatal(msg, UnitType)), Fatal(msg, UnitType))
     mod = Module()
     p = Prelude(mod)
     orig = Function([], p.hd(p.nil()), TupleType([]))
     assert alpha_equal(dcpe(orig, mod=mod).body, Fatal(relay.NO_MATCH_MSG))
 
 
 if __name__ == '__main__':
-    pytest.main()
+    pytest.main([__file__])