[tvm] branch main updated: [microNPU] Re-enable LayoutOptimizer pass (#9793)

[ma...@apache.org]

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 85e27c1  [microNPU] Re-enable LayoutOptimizer pass (#9793)
85e27c1 is described below

commit 85e27c1cf6d87ff7af68dd940a0de0ae4f286eaa
Author: lhutton1 <35...@users.noreply.github.com>
AuthorDate: Thu Dec 23 11:41:12 2021 +0100

    [microNPU] Re-enable LayoutOptimizer pass (#9793)
    
    * [microNPU] Re-enable LayoutOptimizer pass
    
    It looks like in #9597 the LayoutOptimizer pass was accidentally removed.
    Probably due to a race condition in PR's. Re-enabling this feature.
    
    Change-Id: I4fc16a440f90277c5fcd887715166332af052c6b
    
    * change pass ordering
    
    Change-Id: I6e7a22f46660029bbf4be3deb2be929cecf5d365
    
    Co-authored-by: lukhut01 (generated by with_the_same_user script) <lu...@e127400.cambridge.arm.com>
---
 python/tvm/relay/backend/contrib/ethosu/codegen.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/tvm/relay/backend/contrib/ethosu/codegen.py b/python/tvm/relay/backend/contrib/ethosu/codegen.py
index 22f248b..0884b24 100644
--- a/python/tvm/relay/backend/contrib/ethosu/codegen.py
+++ b/python/tvm/relay/backend/contrib/ethosu/codegen.py
@@ -295,6 +295,7 @@ def relay_to_tir_func(ext_func: relay.Function) -> tvm.tir.PrimFunc:
     mod["main"] = ext_func
     mod = LegalizeEthosU()(mod)
     mod = LUTsOptimizer()(mod)
+    mod = LayoutOptimizer()(mod)
     mod = relay.transform.InferType()(mod)
     # We are currently using copy_constants scheduler In the long run,
     # this should be a single intelligent and a composite scheduler