[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