1 files changed, 1 insertions, 6 deletions
diff --git a/mlir/lib/Analysis/SliceAnalysis.cpp b/mlir/lib/Analysis/SliceAnalysis.cpp
index 2b1cf411ceee..45053347730a 100644
--- a/mlir/lib/Analysis/SliceAnalysis.cpp
+++ b/mlir/lib/Analysis/SliceAnalysis.cpp
@@ -132,12 +132,7 @@ void mlir::getBackwardSlice(Operation *op,
 
 void mlir::getBackwardSlice(Value root, SetVector<Operation *> *backwardSlice,
                             const BackwardSliceOptions &options) {
-  if (Operation *definingOp = root.getDefiningOp()) {
-    getBackwardSlice(definingOp, backwardSlice, options);
-    return;
-  }
-  Operation *bbAargOwner = cast<BlockArgument>(root).getOwner()->getParentOp();
-  getBackwardSlice(bbAargOwner, backwardSlice, options);
+  getBackwardSlice(root.getOwningOp(), backwardSlice, options);
 }
 
 SetVector<Operation *>