[PATCH] D141923: [mlir][vector] Add a custom builder for LowerVectorsOp
Quentin Colombet via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Tue Jan 17 05:36:21 PST 2023
qcolombet updated this revision to Diff 489782.
qcolombet added a comment.
- Make sure to match the default values of LowerVectorsOp with what is set in the td file (thanks @nicolasvasilache for noticing)
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D141923/new/
https://reviews.llvm.org/D141923
Files:
mlir/include/mlir/Dialect/Vector/TransformOps/VectorTransformOps.h
mlir/include/mlir/Dialect/Vector/TransformOps/VectorTransformOps.td
Index: mlir/include/mlir/Dialect/Vector/TransformOps/VectorTransformOps.td
===================================================================
--- mlir/include/mlir/Dialect/Vector/TransformOps/VectorTransformOps.td
+++ mlir/include/mlir/Dialect/Vector/TransformOps/VectorTransformOps.td
@@ -46,6 +46,18 @@
);
let results = (outs PDL_Operation:$results);
+ let builders = [
+ OpBuilder<(ins "Type":$resultType, "Value":$target,
+ "const vector::LowerVectorsOptions &":$options), [{
+ return build($_builder, $_state, resultType, target,
+ options.vectorContractLowering,
+ options.vectorMultiReductionLowering, options.vectorTransferSplit,
+ options.vectorTransposeLowering, options.transposeAVX2Lowering,
+ options.unrollVectorTransfers);
+ }]
+ >
+ ];
+
let assemblyFormat = [{
$target
oilist (
Index: mlir/include/mlir/Dialect/Vector/TransformOps/VectorTransformOps.h
===================================================================
--- mlir/include/mlir/Dialect/Vector/TransformOps/VectorTransformOps.h
+++ mlir/include/mlir/Dialect/Vector/TransformOps/VectorTransformOps.h
@@ -11,12 +11,14 @@
#include "mlir/Dialect/PDL/IR/PDLTypes.h"
#include "mlir/Dialect/Transform/IR/TransformInterfaces.h"
+#include "mlir/Dialect/Vector/Transforms/VectorRewritePatterns.h"
#include "mlir/Dialect/Vector/Transforms/VectorTransforms.h"
#include "mlir/IR/OpImplementation.h"
namespace mlir {
namespace vector {
class VectorOp;
+struct LowerVectorsOptions;
} // namespace vector
} // namespace mlir
@@ -32,6 +34,30 @@
namespace vector {
void registerTransformDialectExtension(DialectRegistry ®istry);
+
+/// Helper structure used to hold the different options of LowerVectorsOp.
+struct LowerVectorsOptions : public VectorTransformsOptions {
+ // Have the default values match the LowerVectorsOp values in the td file.
+ LowerVectorsOptions() : VectorTransformsOptions() {
+ setVectorTransformsOptions(VectorContractLowering::OuterProduct);
+ setVectorMultiReductionLowering(
+ VectorMultiReductionLowering::InnerParallel);
+ setVectorTransposeLowering(VectorTransposeLowering::EltWise);
+ setVectorTransferSplit(VectorTransferSplit::LinalgCopy);
+ }
+
+ bool transposeAVX2Lowering = false;
+ LowerVectorsOptions &setTransposeAVX2Lowering(bool opt) {
+ transposeAVX2Lowering = opt;
+ return *this;
+ }
+
+ bool unrollVectorTransfers = true;
+ LowerVectorsOptions &setUnrollVectorTransfers(bool opt) {
+ unrollVectorTransfers = opt;
+ return *this;
+ }
+};
} // namespace vector
} // namespace mlir
-------------- next part --------------
A non-text attachment was scrubbed...
Name: D141923.489782.patch
Type: text/x-patch
Size: 2637 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20230117/c9cbd9f1/attachment.bin>
More information about the llvm-commits
mailing list