diff options
| author | Charitha Saumya <136391709+charithaintc@users.noreply.github.com> | 2024-04-18 11:13:49 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-18 21:13:49 +0300 |
| commit | c577f91d266b74d1b5df475fa2dce7c47fc6c57e (patch) | |
| tree | 3d335f08983f379005927aa9e5a1882eb6d89949 /mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | |
| parent | 44713f15f9116a83f168f420ebddc6ae061883c0 (diff) | |
[mlir][vector] Add support for linearizing Extract, ExtractStridedSlice, Shuffle VectorOps in VectorLinearize (#88204)
This PR adds support for converting `vector.extract_strided_slice` and
`vector.extract` operations to equivalent `vector.shuffle` operations
that operates on linearized (1-D) vectors. `vector.shuffle` operations
operating on n-D (n > 1) are also converted to equivalent shuffle
operations working on linearized vectors.
Diffstat (limited to 'mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp')
| -rw-r--r-- | mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp index 006225999105..c978699e179f 100644 --- a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp +++ b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp @@ -867,6 +867,8 @@ struct TestVectorLinearize final vector::populateVectorLinearizeTypeConversionsAndLegality( typeConverter, patterns, target, targetVectorBitwidth); + vector::populateVectorLinearizeShuffleLikeOpsPatterns( + typeConverter, patterns, target, targetVectorBitwidth); if (failed(applyPartialConversion(getOperation(), target, std::move(patterns)))) return signalPassFailure(); |