Skip to content

Commit 9b36aca

Browse files
committed
Fix Python test
1 parent 060f098 commit 9b36aca

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

mlir/test/python/dialects/transform_vector_ext.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -74,9 +74,9 @@ def enum_configurable_patterns():
7474
# CHECK: transform.apply_patterns.vector.lower_contraction
7575
vector.ApplyLowerContractionPatternsOp()
7676
# CHECK: transform.apply_patterns.vector.lower_contraction
77-
# CHECK-SAME: lowering_strategy = matmulintrinsics
77+
# CHECK-SAME: lowering_strategy = llvm
7878
vector.ApplyLowerContractionPatternsOp(
79-
lowering_strategy=vector.VectorContractLowering.Matmul
79+
lowering_strategy=vector.VectorContractLowering.LLVM
8080
)
8181
# CHECK: transform.apply_patterns.vector.lower_contraction
8282
# CHECK-SAME: lowering_strategy = parallelarith
@@ -105,9 +105,9 @@ def enum_configurable_patterns():
105105
lowering_strategy=vector.VectorTransposeLowering.EltWise
106106
)
107107
# CHECK: transform.apply_patterns.vector.lower_transpose
108-
# CHECK-SAME: lowering_strategy = flat_transpose
108+
# CHECK-SAME: lowering_strategy = llvm
109109
vector.ApplyLowerTransposePatternsOp(
110-
lowering_strategy=vector.VectorTransposeLowering.Flat
110+
lowering_strategy=vector.VectorTransposeLowering.LLVM
111111
)
112112
# CHECK: transform.apply_patterns.vector.lower_transpose
113113
# CHECK-SAME: lowering_strategy = shuffle_1d
@@ -120,10 +120,10 @@ def enum_configurable_patterns():
120120
lowering_strategy=vector.VectorTransposeLowering.Shuffle16x16
121121
)
122122
# CHECK: transform.apply_patterns.vector.lower_transpose
123-
# CHECK-SAME: lowering_strategy = flat_transpose
123+
# CHECK-SAME: lowering_strategy = llvm
124124
# CHECK-SAME: avx2_lowering_strategy = true
125125
vector.ApplyLowerTransposePatternsOp(
126-
lowering_strategy=vector.VectorTransposeLowering.Flat,
126+
lowering_strategy=vector.VectorTransposeLowering.LLVM,
127127
avx2_lowering_strategy=True,
128128
)
129129

0 commit comments

Comments
 (0)