[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