[Mlir-commits] [mlir] [mlir][vector] Make the in_bounds attribute mandatory (PR #97049)
Diego Caballero
llvmlistbot at llvm.org
Thu Jul 11 09:34:56 PDT 2024
================
@@ -155,6 +155,13 @@ class AffineMap {
bool isMinorIdentityWithBroadcasting(
SmallVectorImpl<unsigned> *broadcastedDims = nullptr) const;
+ /// Returns the list of broadcast dimensions.
+ /// Ex:
+ /// * (d0, d1, d2) -> (0, d1) gives [0]
+ /// * (d0, d1, d2) -> (d2, d1) gives []
+ /// * (d0, d1, d2, d4) -> (d0, 0, d1, 0) gives [1, 3]
+ SmallVector<unsigned> getBroadcastDims() const;
----------------
dcaballe wrote:
Have you looked at `isMinorIdentityWithBroadcasting`
https://github.com/llvm/llvm-project/pull/97049
More information about the Mlir-commits
mailing list