[CINN] ComputeAtReductionTactic supports elementwise-to-reduce fusion

Author: lshpku

paddle/cinn/ir/group_schedule/dy_shape_group_scheduler.cc

@@ -24,6 +24,8 @@
 #include "paddle/cinn/ir/op/ir_operators.h"
 #include "paddle/common/enforce.h"
 
+PD_DECLARE_bool(cinn_enable_compute_at);
+
 namespace cinn {
 namespace ir {
 
@@ -36,7 +38,9 @@ void DynamicShapeGroupScheduler::Init() {
   tactics_.emplace_back(CreateTileBroadcastTactic());
   tactics_.emplace_back(CreateTileFirstGeneralTactic());
   tactics_.emplace_back(CreateComputeInlineTactic());
-  tactics_.emplace_back(CreateComputeAtReductionTactic());
+  if (FLAGS_cinn_enable_compute_at) {
+    tactics_.emplace_back(CreateComputeAtReductionTactic());
+  }
 }
 
 void DynamicShapeGroupScheduler::InitBuckets() {