[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