diff --git a/paddle/pir/src/dialect/shape/transforms/shape_optimization_pass.cc b/paddle/pir/src/dialect/shape/transforms/shape_optimization_pass.cc index 28cf35696e96f..938f463beeb27 100644 --- a/paddle/pir/src/dialect/shape/transforms/shape_optimization_pass.cc +++ b/paddle/pir/src/dialect/shape/transforms/shape_optimization_pass.cc @@ -347,6 +347,7 @@ static const std::set skip_cache_check_op_set = { "pd_op.arange", "pd_op.masked_select", "pd_op.sync_batch_norm_", + "pd_op.slice", // unneeded to cache "cinn_op.generate_shape", };