[Mlir-commits] [mlir] [mlir][linalg] Enable masked vectorisation for depthwise convolutions (PR #81625)
Andrzej WarzyĆski
llvmlistbot at llvm.org
Tue Mar 5 09:50:18 PST 2024
================
@@ -3268,7 +3358,9 @@ struct Conv1DGenerator
/// Entry point that transposes into the common form:
/// {{n, strideW * w + dilationW * kw, c}, {kw, c}, {n, w, c}}
- FailureOr<Operation *> generateDilatedConv(bool flatten = false) {
+ FailureOr<Operation *> generateDilatedConv(uint64_t vecChDimSize = 0,
+ bool vecChDimScalableFlag = true,
----------------
banach-space wrote:
Leftover from experiments, thanks for pointing out!
https://github.com/llvm/llvm-project/pull/81625
More information about the Mlir-commits
mailing list