-
Notifications
You must be signed in to change notification settings - Fork 70
Open
Labels
Description
Motivation: Expanding Tuner Support Beyond NHWC_HWCF Layout
Currently, the tuner only supports the NHWC_HWCF layout. However, there exists one demand from the BOO to support additional layouts. IREE already can handle more layouts through the TileAndFuse pipeline with the IGEMM option (via setIGEMMConvolutionLoweringConfig).
To extend tuner support, we will proceed with the following approach:
Implementation Plan
- Expose the necessary Python bindings to leverage IREE and LLVM capabilities
[Codegen][Tuner] expose python binding for getIGEMMGenericConvDetails iree-org/iree#22598
[mlir][linalg][python] Add Python Bindings for Inferring Contraction Dimensions from Affine Maps llvm/llvm-project#167587 - Ensure that the existing convolution tuner integrates properly with the new Python bindings
[tuner] use igemm bindings #2683 - Add padding support for unaligned cases (note: there are two existing attributes, "padding" and "padding_conv", that need to be reconciled)
- Implement support for additional layout configurations
- Refactor and remove legacy code specific to single-layout implementations