[USMP] Fix assert condition for TVMBackendAllocWorkspace (apache#11270)

* Fix test condition * fix
Deelvin · May 17, 2022 · 62d8c18 · 62d8c18
1 parent 0edb7c5
commit 62d8c18
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/python/relay/aot/test_crt_aot_usmp.py b/tests/python/relay/aot/test_crt_aot_usmp.py
@@ -54,9 +54,9 @@ def check_for_no_tvm_backendallocworkspace_calls(mod: tvm.runtime.module):
             dso_mod.type_key == "c"
         ), 'Current CRT AoT codegen flow should only produce type "c" runtime modules'
         source = dso_mod.get_source()
-        source.count(
-            "TVMBackendAllocWorkspace"
-        ) == 0, "This is failing because USMP was unable to plan for every tir.allocate node"
+        assert (
+            source.count("TVMBackendAllocWorkspace") == 0
+        ), "This is failing because USMP was unable to plan for every tir.allocate node"
 
 
 @pytest.mark.parametrize(