kavin-sai-krishna commented on issue #17890: URL: https://github.com/apache/tvm/issues/17890#issuecomment-2875038869
Hi, Sure, here's the updated version for `conv2d_transpose`: --- I've been debugging the issue and found that the legalization pass for `conv2d_transpose` fails with the following error: > "TOPI conv2d\_transpose does not support input layout other than NCHW, and kernel layout other than IOHW, so cannot be legalized by TOPI" This error comes from `tvm/python/tvm/relax/transform/legalize_ops/nn.py`. It seems the kernel layout provided by PyTorch don't match what TOPI expects. To resolve this, I see two potential options: 1. **Insert layout transformations in the frontend** using `R.op.nn.layout_transform` to convert the kernel to `IOHW` before legalization. 2. **Implement the `InferLayout` function** for `conv2d_transpose` in `tvm/src/relax/op/nn/convolution.cc`, which is currently unimplemented. @Hzfengsy @tlopex @mshr-h Would appreciate any suggestions or feedback on the best way to move forward. -- 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: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
