From 95102060045bb745becd509c833dc4cbecd28e0a Mon Sep 17 00:00:00 2001
From: Jerry Wu <cheyuw@google.com>
Date: Fri, 11 Aug 2023 17:58:11 +0000
Subject: [PATCH] Fix ASAN

---
 .../iree/compiler/Codegen/Common/DecomposeBatchMmt4DOps.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/compiler/src/iree/compiler/Codegen/Common/DecomposeBatchMmt4DOps.cpp b/compiler/src/iree/compiler/Codegen/Common/DecomposeBatchMmt4DOps.cpp
index 1a23d02f32b7..0863ede83dc1 100644
--- a/compiler/src/iree/compiler/Codegen/Common/DecomposeBatchMmt4DOps.cpp
+++ b/compiler/src/iree/compiler/Codegen/Common/DecomposeBatchMmt4DOps.cpp
@@ -60,12 +60,14 @@ struct ConvertBatchMmt4DtoMmt4DPattern
     }
 
     auto lhsType = lhs.getType().cast<RankedTensorType>();
-    auto reducedLhsType = RankedTensorType::Builder(lhsType).dropDim(0);
+    RankedTensorType reducedLhsType =
+        RankedTensorType::Builder(lhsType).dropDim(0);
     auto reducedLhs = tensor::createCanonicalRankReducingExtractSliceOp(
         rewriter, loc, lhs, reducedLhsType);
 
     auto rhsType = rhs.getType().cast<RankedTensorType>();
-    auto reducedRhsType = RankedTensorType::Builder(rhsType).dropDim(0);
+    RankedTensorType reducedRhsType =
+        RankedTensorType::Builder(rhsType).dropDim(0);
     auto reducedRhs = tensor::createCanonicalRankReducingExtractSliceOp(
         rewriter, loc, rhs, reducedRhsType);