[Mlir-commits] [mlir] [mlir][Vector] Improve support for vector.extract(broadcast) (PR #116234)

Diego Caballero llvmlistbot at llvm.org
Thu Nov 14 10:15:30 PST 2024


================
@@ -1680,6 +1676,16 @@ static Value foldExtractFromBroadcast(ExtractOp extractOp) {
           broadcastVecType.getShape().take_back(extractResultRank))
     return Value();
 
+  // The dim-1 broadcast -> ExtractOp folder requires in place operation
----------------
dcaballe wrote:

I think I know what you mean by "in-place operation" but it would be good to illustrate that with an example... I think there are cases that could be supported...

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


More information about the Mlir-commits mailing list