Skip to content

Commit 796a1cf

Browse files
authored
[mlir][nfc] Update vectorize-tensor-extract.mlir (2/N) (#119080)
Tests in "vectorize-tensor-extract.mlir" are inconsistent and would benefit from refactoring to: * Clearly categorize tests into "contiguous load," "gather load," and "scalar load + broadcast" cases, reflecting the structure of tensor.extract vectorization. * Unify variable naming (both MLIR and FileCheck). * Ensure all tests exercise unmasked vectorization (masked vectorization is covered in "vectorize-tensor-extract-masked.mlir"). * Improve and standardize formatting. These changes will make it easier to identify the test cases being exercised and simplify future maintenance or refactoring. This is patch 2/N in the series; below is a summary of the specific changes in this patch. ---------------------------------------------------------------------- Summary of patch 2/N ---------------------------------------------------------------------- Makes all tests re-use the same TD sequence. This TD sequence has been extracted to a deducted file: * "td/vectorize-with-patterns.mlir". ---------------------------------------------------------------------- Previous patches: ---------------------------------------------------------------------- * #118977
1 parent 1bc1703 commit 796a1cf

File tree

2 files changed

+28
-196
lines changed

2 files changed

+28
-196
lines changed
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
module @transforms attributes { transform.with_named_sequence } {
2+
transform.named_sequence @vectorize_with_patterns(%module: !transform.any_op {transform.readonly}) {
3+
4+
%0 = transform.structured.match ops{["linalg.generic"]} in %module : (!transform.any_op) -> !transform.any_op
5+
%1 = transform.get_parent_op %0 {isolated_from_above} : (!transform.any_op) -> !transform.any_op
6+
%2 = transform.structured.vectorize_children_and_apply_patterns %1 { vectorize_nd_extract } : (!transform.any_op) -> !transform.any_op
7+
8+
transform.yield
9+
}
10+
}

0 commit comments

Comments
 (0)