PaddlePaddle
diff --git a/‎paddle/cinn/ir/group_schedule/dy_shape_group_scheduler.cc
Copy file name to clipboard
+3Lines changed: 3 additions & 0 deletions b/‎paddle/cinn/ir/group_schedule/dy_shape_group_scheduler.cc
Copy file name to clipboard
+3Lines changed: 3 additions & 0 deletions
diff --git a/‎paddle/cinn/ir/group_schedule/tactic/CMakeLists.txt
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions b/‎paddle/cinn/ir/group_schedule/tactic/CMakeLists.txt
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
@@ -15,6 +15,7 @@
 #include "paddle/cinn/ir/group_schedule/dy_shape_group_scheduler.h"
 #include "paddle/cinn/common/cas.h"
 #include "paddle/cinn/ir/group_schedule/config/schedule_config_manager.h"
+#include "paddle/cinn/ir/group_schedule/tactic/compute_at_reduction_tactic.h"
 #include "paddle/cinn/ir/group_schedule/tactic/compute_inline_tactic.h"
 #include "paddle/cinn/ir/group_schedule/tactic/tile_first_general_tactic.h"
 #include "paddle/cinn/ir/ir_analyzer/ir_analyzer.h"
@@ -36,6 +37,8 @@ void DynamicShapeGroupScheduler::Init() {
   VLOG(4) << "CreateTileFirstGeneralTactic End";
   tactics_.emplace_back(CreateComputeInlineTactic());
   VLOG(4) << "CreateTileCreateComputeInlineTactic End";
+  tactics_.emplace_back(CreateComputeAtReductionTactic());
+  VLOG(4) << "CreateComputeAtReductionTactic End";
 }
 
 void DynamicShapeGroupScheduler::InitBuckets() {
 
@@ -4,6 +4,7 @@ gather_srcs(cinnapi_src SRCS align_iter_space_tactic.cc)
 gather_srcs(cinnapi_src SRCS tile_tactic.cc)
 gather_srcs(cinnapi_src SRCS compute_inline_tactic.cc)
 gather_srcs(cinnapi_src SRCS optimize_reduction_tactic.cc)
+gather_srcs(cinnapi_src SRCS compute_at_reduction_tactic.cc)
 gather_srcs(cinnapi_src SRCS bind_cuda_tactic.cc)
 gather_srcs(cinnapi_src SRCS arrange_storage_tactic.cc)
 gather_srcs(cinnapi_src SRCS tile_first_general_tactic.cc)