[3/10] Moved TIR generation from Python to C++ for CMSIS-NN (apache#8951

) * [CMSIS-NN] Moved TIR Generation to C++ * Deleted self import for cmsisnn Change-Id: I2cdcd7a90aa4749877c48bc6c7c4d27328856860 * Reusing CodeGenC VistiExpr for softmax Change-Id: Ie41b695fa06468cd3b0bfe428c360e98438a9180
ylc · Jan 13, 2022 · e67536a · e67536a
1 parent dcaf8c9
commit e67536a
Show file tree

Hide file tree

Showing 8 changed files with 306 additions and 353 deletions.
diff --git a/python/tvm/relay/backend/__init__.py b/python/tvm/relay/backend/__init__.py
@@ -16,4 +16,3 @@
 # under the License.
 """Backend codegen modules for relay."""
 from . import compile_engine
-from .contrib import cmsisnn
diff --git a/python/tvm/relay/backend/contrib/__init__.py b/python/tvm/relay/backend/contrib/__init__.py
diff --git a/python/tvm/relay/backend/contrib/cmsisnn/__init__.py b/python/tvm/relay/backend/contrib/cmsisnn/__init__.py
diff --git a/python/tvm/relay/backend/contrib/cmsisnn/codegen.py b/python/tvm/relay/backend/contrib/cmsisnn/codegen.py
diff --git a/python/tvm/relay/op/contrib/cmsisnn.py b/python/tvm/relay/op/contrib/cmsisnn.py
@@ -80,5 +80,5 @@ def check_quantized_softmax(extract):
         )
 
     return [
-        ("cmsisnn.qnn_softmax", softmax_pattern(), check_quantized_softmax),
+        ("cmsisnn.quantized_softmax", softmax_pattern(), check_quantized_softmax),
     ]