[Mlir-commits] [mlir] [mlir][Linalg] Refine how broadcast dims are treated (PR #99015)

[Diego Caballero]

================
@@ -346,6 +346,10 @@ class AffineMap {
   /// returns the resulting values. `this` must be symbol-less.
   SmallVector<int64_t, 4> compose(ArrayRef<int64_t> values) const;
 
+  size_t numOfZeroResults() const;
+
+  AffineMap dropZeros();
----------------
dcaballe wrote:

If there is nothing that provides what you need I would just introduce what you need and try to reuse/refactor at implementation level.

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