From 8dd1bf9ed385d497a9238a190a6a2fb67507947d Mon Sep 17 00:00:00 2001
From: MaheshRavishankar <mravisha@smc300x-clt-r4c4-02.cs-clt.dcgpu>
Date: Fri, 10 Jan 2025 15:00:10 -0600
Subject: [PATCH] Propagate expand_shape past collapse_shape

Signed-off-by: MaheshRavishankar <mravisha@smc300x-clt-r4c4-02.cs-clt.dcgpu>
---
 .../src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/compiler/src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp b/compiler/src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp
index 48ee4d4d80c1..dc9f855ad4eb 100644
--- a/compiler/src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp
+++ b/compiler/src/iree/compiler/DispatchCreation/BubbleUpExpandShapes.cpp
@@ -204,6 +204,7 @@ void BubbleUpExpandShapesPass::runOnOperation() {
   // that can be done later) of reshape ops.
   tensor::populateFoldTensorEmptyPatterns(bubbleExpandShapePatterns);
   bubbleExpandShapePatterns.insert<BubbleExpandThroughExtract>(context);
+  tensor::populateBubbleUpExpandShapePatterns(bubbleExpandShapePatterns);
   tensor::ExpandShapeOp::getCanonicalizationPatterns(bubbleExpandShapePatterns,
                                                      context);