[Mlir-commits] [mlir] [MLIR][Shape] Support >2 args in `shape.broadcast` folder (PR #126808)

[Mateusz Sokół]

================
@@ -86,6 +86,19 @@ func.func @broadcast() -> !shape.shape {
 
 // -----
 
+// Variadic case including extent tensors.
----------------
mtsokol wrote:

One and two operands input is already supported and tested. Here's an existing test for single operand:
https://github.com/llvm/llvm-project/blob/e7bf54d62771219145171c66584578972edf0e30/mlir/test/Dialect/Shape/canonicalize.mlir#L1347-L1354

My tests adds >2 input case.

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