From 3849574600dd763b9488f5c5cac1681bd8132435 Mon Sep 17 00:00:00 2001
From: Ian Wood <ianwood2024@u.northwestern.edu>
Date: Thu, 5 Dec 2024 01:51:36 -0800
Subject: [PATCH] Small cleanup

Signed-off-by: Ian Wood <ianwood2024@u.northwestern.edu>
---
 .../iree/compiler/DispatchCreation/FoldUnitExtentDims.cpp    | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/compiler/src/iree/compiler/DispatchCreation/FoldUnitExtentDims.cpp b/compiler/src/iree/compiler/DispatchCreation/FoldUnitExtentDims.cpp
index ebc34423ccd3..fa5fb6bf13af 100644
--- a/compiler/src/iree/compiler/DispatchCreation/FoldUnitExtentDims.cpp
+++ b/compiler/src/iree/compiler/DispatchCreation/FoldUnitExtentDims.cpp
@@ -23,7 +23,6 @@
 #include "mlir/Dialect/Linalg/Transforms/Transforms.h"
 #include "mlir/Dialect/Tensor/IR/Tensor.h"
 #include "mlir/Dialect/Utils/ReshapeOpsUtils.h"
-#include "mlir/IR/Verifier.h"
 #include "mlir/Pass/Pass.h"
 #include "mlir/Transforms/GreedyPatternRewriteDriver.h"
 
@@ -115,11 +114,11 @@ struct DropUnitDimsFromCollapseOfExpand
         };
 
     auto dropOutputOfr = [&toDrop](const SmallVector<OpFoldResult> &sizes) {
-      return llvm::to_vector(llvm::map_range(
+      return llvm::map_to_vector(
           llvm::make_filter_range(
               llvm::enumerate(sizes),
               [&toDrop](auto pair) { return !toDrop.contains(pair.index()); }),
-          [](auto pair) -> OpFoldResult { return pair.value(); }));
+          [](auto pair) -> OpFoldResult { return pair.value(); });
     };
 
     auto isIdentityReassociation = [](ArrayRef<ReassociationIndices> reassoc) {