[Mlir-commits] [mlir] [mlir][ArmSME] Lower extract from 2D scalable create_mask to psel (PR #96066)
Andrzej Warzyński
llvmlistbot at llvm.org
Wed Jun 19 12:43:31 PDT 2024
================
@@ -1320,3 +1320,35 @@ func.func @vector_extract_element_f64(%row: index, %col: index) -> f64 {
%el = vector.extract %tile[%row, %col] : f64 from vector<[2]x[2]xf64>
return %el : f64
}
+
+// -----
+
+// CHECK-LABEL: @dynamic_vector_extract_mask_to_psel(
+// CHECK-SAME: %[[A:[a-z0-9]+]]: index,
+// CHECK-SAME: %[[B:[a-z0-9]+]]: index,
+// CHECK-SAME: %[[INDEX:[a-z0-9]+]]: index)
----------------
banach-space wrote:
This would work:
```
// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index, %[[INDEX:.*]]: index)
```
Or just update the other regex'es in the file, there's not that many 🤔 Just a nit.
https://github.com/llvm/llvm-project/pull/96066
More information about the Mlir-commits
mailing list