[tvm] branch main updated: [USMP] Fix assert condition for TVMBackendAllocWorkspace (#11270)

[manupa]

This is an automated email from the ASF dual-hosted git repository.

manupa pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 391b4d99ab [USMP] Fix assert condition for TVMBackendAllocWorkspace 
(#11270)
391b4d99ab is described below

commit 391b4d99ab6d9e0799e5b060ea42893240876be7
Author: Mehrdad Hessar <mhes...@octoml.ai>
AuthorDate: Wed May 11 01:50:40 2022 -0700

    [USMP] Fix assert condition for TVMBackendAllocWorkspace (#11270)
    
    * Fix test condition
    
    * fix
---
 tests/python/relay/aot/test_crt_aot_usmp.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/python/relay/aot/test_crt_aot_usmp.py 
b/tests/python/relay/aot/test_crt_aot_usmp.py
index 86e0d18021..ab7fb4167c 100644
--- 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(