Skip to content

Commit

Permalink
[microNPU] Re-enable LayoutOptimizer pass (apache#9793)
Browse files Browse the repository at this point in the history
* [microNPU] Re-enable LayoutOptimizer pass

It looks like in apache#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) <lukhut01@e127400.cambridge.arm.com>
  • Loading branch information
2 people authored and ylc committed Jan 13, 2022
1 parent 5842335 commit ccfd8dd
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions python/tvm/relay/backend/contrib/ethosu/codegen.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit ccfd8dd

Please sign in to comment.