Fix ASAN

iree-org · Aug 11, 2023 · 9510206 · 9510206
1 parent eda8687
commit 9510206
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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);