lhutton1 opened a new issue, #16536: URL: https://github.com/apache/tvm/issues/16536
### Expected behaviour: When an `arm_cpu` target is used, the model should compile successfully without an error. ### Actual behaviour: When compiled on an `arm_cpu` target, the models results in the following error: ``` python/tvm/autotvm/task/topi_integration.py:165: in wrapper node = topi_compute(cfg, *args) python/tvm/topi/arm_cpu/conv2d.py:50: in conv2d_nchw_spatial_pack return conv2d_spatial_pack_nchw( def conv2d_spatial_pack_nchw(cfg, data, kernel, strides, padding, dilation, out_dtype, num_tile): """compute define for Conv2d Spatial Pack with NCHW layout""" out_dtype = out_dtype or data.dtype N, CI, IH, IW = get_const_tuple(data.shape) if isinstance(N, tvm.tir.Any): N = tvm.te.size_var("n") if not isinstance(IH, int) or not isinstance(IW, int): > raise RuntimeError("ARM winograd conv2d doesn't support dynamic input height or width.") E RuntimeError: ARM winograd conv2d doesn't support dynamic input height or width. ``` ### Environment: Tested with TVM at https://github.com/apache/tvm/commit/6a3fadc0654ecf9557ffe08d24677684c96e80b0. The issue was found as a result of the changes in https://github.com/apache/tvm/pull/16513, however it can be reproduced without as described below. ### How to reproduce: Run the following tests: - pytest tests/python/relay/test_any.py -k test_any_conv2d with an `arm_cpu` target. Note: Reminder to remove any skip condition that exists in the test currently. ----------- Likely schedule selection in `relay/strategy/arm_cpu.py` needs to be fixed not to select a schedule that doesn't support dynamic height/width. -- 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. To unsubscribe, e-mail: commits-unsubscr...@tvm.apache.org.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org