From fe97c33b301435edb5f0a702f355eb96021c9914 Mon Sep 17 00:00:00 2001 From: Andrey Malyshev Date: Wed, 25 Aug 2021 14:53:47 +0300 Subject: [PATCH] Fix get_graph_json test --- tests/python/unittest/test_runtime_module_based_interface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_runtime_module_based_interface.py b/tests/python/unittest/test_runtime_module_based_interface.py index 9eeea41503be..90b8b1b20cd2 100644 --- 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