[Mlir-commits] [mlir] [mlir][spirv] Enable dot operation for bfloat16 (PR #145409)

[Jakub Kuderski]

https://github.com/kuhar commented:

> Is there a way to require extension/capability/availability for ops that use a specific type? In this case for example using dot with bf16 requires the `BFloat16DotProductKHR` capability.

Yes, I think we need to update: https://github.com/llvm/llvm-project/blob/cfdc4c4a5b671646cb08aeab106103cc7006cb89/mlir/lib/Dialect/SPIRV/IR/IntegerDotProductOps.cpp#L92. The tests are in https://github.com/llvm/llvm-project/blob/cfdc4c4a5b671646cb08aeab106103cc7006cb89/mlir/test/Dialect/SPIRV/IR/availability.mlir#L54

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