diff --git a/mlir/lib/Conversion/BufferizationToMemRef/BufferizationToMemRef.cpp b/mlir/lib/Conversion/BufferizationToMemRef/BufferizationToMemRef.cpp
--- a/mlir/lib/Conversion/BufferizationToMemRef/BufferizationToMemRef.cpp
+++ b/mlir/lib/Conversion/BufferizationToMemRef/BufferizationToMemRef.cpp
@@ -98,7 +98,9 @@
     ConversionTarget target(getContext());
     target.addLegalDialect<memref::MemRefDialect>();
     target.addLegalOp<arith::ConstantOp>();
-    target.addIllegalDialect<bufferization::BufferizationDialect>();
+    // In the current state, This pass only transforms a CloneOp in
+    // Bufferization dialect.
+    target.addIllegalOp<bufferization::CloneOp>();
 
     if (failed(applyPartialConversion(getOperation(), target,
                                       std::move(patterns))))