[Mlir-commits] [mlir] [mlir][vector] Allow 0-D vector acc/res in `vector.contract` (PR #234638)
Federico Bruzzone
llvmlistbot at llvm.org
Thu Oct 22 23:26:39 PDT 2026
================
@@ -72,7 +72,8 @@ def Vector_ContractionOp :
intermediate result to the accumulator argument of rank K, and returns a
vector result of rank K (where K = num_lhs_free_dims + num_rhs_free_dims +
num_batch_dims (see dimension type descriptions below)). For K = 0 (no
- free or batch dimensions), the accumulator and output are a scalar.
+ free or batch dimensions), the accumulator and output are either both a
+ scalar or both a 0-D vector.
----------------
FedericoBruzzone wrote:
You're right. Thanks!
https://github.com/llvm/llvm-project/pull/234638
More information about the Mlir-commits
mailing list