[Mlir-commits] [mlir] [mlir][vector] Bugfix of linearize `vector.extract` (PR #106836)

Cullen Rhodes llvmlistbot at llvm.org
Wed Sep 4 00:14:23 PDT 2024


================
@@ -306,3 +306,14 @@ func.func @test_vector_insert_scalable(%arg0: vector<2x8x[4]xf32>, %arg1: vector
   // ALL: return %[[RES]] : vector<2x8x[4]xf32>
   return %0 : vector<2x8x[4]xf32>
 }
+
+// -----
+
+// ALL-LABEL: test_vector_extract_scalar
+func.func @test_vector_extract_scalar() {
+  %cst = arith.constant dense<[1, 2, 3, 4]> : vector<4xi32>
+  // ALL:     vector.extract
+  // ALL-NOT: vector.shuffle
----------------
c-rhodes wrote:

this would still pass if a shuffle came before `vector.extract`, have to check both sides with CHECK-NOT 
```suggestion
  // ALL-NOT: vector.shuffle
  // ALL:     vector.extract
  // ALL-NOT: vector.shuffle
```

https://github.com/llvm/llvm-project/pull/106836


More information about the Mlir-commits mailing list