crazydemo opened a new pull request #10421: URL: https://github.com/apache/tvm/pull/10421
This PR aims to support BYOC-DNNL run in the dnnl optimal layout. There are two changes needed to be noticed. - For group conv with dnnl backend, its layout need to be legalized to `GOIHW` first, so that it can run in optimal dnnl layout like HWOIG16g. Changes in `Convolution.cc` is needed to enable group conv run in GOIHW layout. - `get_optimal_layout_for_conv` and `get_optimal_layout_for_deconv` functions are registered in `tvm.relay.contrib` to query the optimal dnnl layout. The related test cases has been added as well. -- 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]
