[Mlir-commits] [mlir] TosaToLinalgNamed: add option to prefer HWCF kernel layout for Conv2D ops. (PR #70482)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Fri Oct 27 11:38:41 PDT 2023


================
@@ -45,8 +48,12 @@ void registerTosaToLinalgPipelines();
 /// Populates conversion passes from TOSA dialect to Linalg dialect.
 void populateTosaToLinalgConversionPatterns(RewritePatternSet *patterns);
 
+enum class Conv2DKernelLayout { FHWC, HWCF };
+
 /// Populates conversion passes from TOSA dialect to Linalg named operations.
-void populateTosaToLinalgNamedConversionPatterns(RewritePatternSet *patterns);
+void populateTosaToLinalgNamedConversionPatterns(
----------------
MaheshRavishankar wrote:

Could we make this take the `populate*` method take `TosaToLinalgNamedOptions` directly? Then we dont need the new `enum`. Also this is a entry point for *all* tosa to linalg named conversion patterns... So we want a stable ABI. Passing the options directly means that more optionality for controlling other named ops dont need API changes.

https://github.com/llvm/llvm-project/pull/70482


More information about the Mlir-commits mailing list