llvm-project
fc81a665 - [NFC][Linalg] Add `matchConvolutionOpOfType` API and make `isaConvolutionOpOfType` API a wrapper (#174722)

Commit
71 days ago
[NFC][Linalg] Add `matchConvolutionOpOfType` API and make `isaConvolutionOpOfType` API a wrapper (#174722) -- This commit involves the following updates pertaining to `isaConvolutionOpOfType` API :- 1. We don't want dilations/strides of convolution op to be returned as pointer arguments to the API function - to tackle this we create a new API `matchConvolutionOpOfType` which would return an optional struct of dilations/stride. 2. To not break the original API's use case as a simple querying functionality with true/false return - we keep `isaConvolutionOpOfType` as a wrapper API which will invoke `matchConvolutionOpOfType` API and return true/false depending on whether `matchConvolutionOpOfType` API returned any value or not. 3. Dilations/strides of named convolution op are also populated now (it was missed in the previous PRs while creating `isaConvolutionOpOfType`). 4. [Max/Min]UnsignedPool ops' body matcher now only matches unsigned int ops (refer: https://github.com/llvm/llvm-project/pull/166070) -- No tests are being added as all the above are NFC changes around the API which already is being tested via Specialize.cpp. Signed-off-by: Abhishek Varma <abhvarma@amd.com>
Parents
Loading