Fix get_graph_json test

apache · Aug 29, 2021 · fe97c33 · fe97c33
1 parent 660a8fb
commit fe97c33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_runtime_module_based_interface.py b/tests/python/unittest/test_runtime_module_based_interface.py
@@ -106,7 +106,7 @@ def test_cpu_get_graph_json():
     loaded_lib = tvm.runtime.load_module(path_lib)
     json = loaded_lib["get_graph_json"]()
     assert isinstance(json, str) == True
-    assert json.find("tvmgen_default_fused_nn_softmax1") == 6312
+    assert json.find("tvmgen_default_fused_nn_softmax1") > -1
 
 
 @tvm.testing.requires_cuda