[Mlir-commits] [mlir] [mlir][spirv] Add support for SPV_ARM_tensors (PR #144667)

Davide Grohmann llvmlistbot at llvm.org
Thu Jun 19 04:52:15 PDT 2025


davidegrohmann wrote:

> Also, we probably should add a test in here: https://github.com/llvm/llvm-project/blob/main/mlir/test/Dialect/SPIRV/IR/availability.mlir

It seems that particular test works only for ops not for types. I have tried to add the following test:
```
// CHECK-LABEL: arm_tensor
func.func @arm_tensor(%0 : !spirv.arm.tensor<1xi32>) -> (!spirv.arm.tensor<1xi32>) {
  // CHECK: min version: v1.6
  // CHECK: max version: v1.6
  // CHECK: extensions: [ [SPV_ARM_tensors] ]
  // CHECK: capabilities: [ [TensorsArm] ]
  spirv.ReturnValue %0 : !spirv.arm.tensor<1xi32>
}
```
it produces the following output:
```
arm_tensor
spirv.ReturnValue min version: v1.0
spirv.ReturnValue max version: v1.6
spirv.ReturnValue extensions: [ ]
spirv.ReturnValue capabilities: [ ]
```

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


More information about the Mlir-commits mailing list