[Mlir-commits] [mlir] [mlir][vector] Add a check to ensure input vector rank equals target shape rank (PR #127706)

Prakhar Dixit llvmlistbot at llvm.org
Thu Feb 20 14:04:46 PST 2025


Prakhar-Dixit wrote:

> Thanks for the updates! Please also add the example from your summary as a negative test.
> 
> You would do it somewhere here:
> 
> https://github.com/llvm/llvm-project/blob/41be5bbbdba2939a5fdb82c968c102f993edc4d8/mlir/test/Dialect/Vector/vector-unroll-options.mlir#L184-L186

I’m a little confused here since this is my first time. Could you help me with it?
Is this what you're expecting : 
```
  func.func @vector_fma_2(%a: vector<24x2x2xf32>) {
    %0 = vector.fma %a, %a, %a : vector<24x2x2xf32>
    %1 = scf.execute_region -> vector<24x2x2xf32> {
        scf.yield %0 : vector<24x2x2xf32>
      }
    return 
  }
// CHECK-LABEL: func @vector_fma_2
//       CHECK: return 
```

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


More information about the Mlir-commits mailing list