mbarrett97 opened a new issue #4542: [AutoTVM] Tuning fails for an NHWC network on Arm CPU URL: https://github.com/apache/incubator-tvm/issues/4542 I'm trying to tune NHWC networks produced by the tflite frontend, but they fail with the error `AssertionError: only support NCHW/HWCN currently`. This suggests to me the AlterOpLayout pass is not happening during task extraction. I can force the AlterOpLayout pass to happen using the following patch: ``` --- a/python/tvm/autotvm/task/relay_integration.py +++ b/python/tvm/autotvm/task/relay_integration.py @@ -49,7 +49,9 @@ def _lower(func, grc = graph_runtime_codegen.GraphRuntimeCodegen(None, target) return grc.codegen(mod["main"]) # default case - mod, _ = relay.optimize(func, target, params) + with relay.build_config(opt_level=3): + mod, _ = relay.optimize(func, target, params) + grc = graph_runtime_codegen.GraphRuntimeCodegen(None, target) return grc.codegen(mod["main"]) ``` which does fix this issue. Am I doing something wrong in AutoTVM or is this patch necessary?
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
