diff --git a/paddle/cinn/ir/group_schedule/config/group_tile_config.cc b/paddle/cinn/ir/group_schedule/config/group_tile_config.cc index cf70a8c933174a..d02f4ef0c2e746 100644 --- a/paddle/cinn/ir/group_schedule/config/group_tile_config.cc +++ b/paddle/cinn/ir/group_schedule/config/group_tile_config.cc @@ -96,7 +96,7 @@ BuildPureStaticShapeConfig( ScheduleConfig::TileConfig tile_config{ /* warp_num = */ warp_num, /* tree_reduce_num = */ 1, - /* spatial_inner_num = */ 1, + /* spatial_inner_num = */ 4, /* reduce_method = */ NoneReduceMethod()}; return {{bucket_info, tile_config}}; } else if (base_info->reduce_numel <= 256) { @@ -227,7 +227,7 @@ BuildStaticReduceConfig( ScheduleConfig::TileConfig tile_config__1024_1M{ /* warp_num = */ 32, /* tree_reduce_num = */ 1, - /* spatial_inner_num = */ 1, + /* spatial_inner_num = */ 4, /* reduce_method = */ NoneReduceMethod()}; BucketInfo bucket_info__1M_INF{/* sp_lower_bound = */ 1024 * 1024, /* sp_upper_bound = */ kMaxNumel, @@ -236,7 +236,7 @@ BuildStaticReduceConfig( ScheduleConfig::TileConfig tile_config__1M_INF{ /* warp_num = */ 32, /* tree_reduce_num = */ 1, - /* spatial_inner_num = */ 16, + /* spatial_inner_num = */ 4, /* reduce_method = */ NoneReduceMethod()}; return {{bucket_info__1_1023, tile_config__1_1023}, {bucket_info__1024_1M, tile_config__1024_1M},