optimize

PaddlePaddle · Apr 7, 2023 · dac0bad · dac0bad
1 parent 78d8de0
commit dac0bad
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/cinn/ir/ir_schedule.cc b/cinn/ir/ir_schedule.cc
@@ -1420,7 +1420,6 @@ Expr ReverseComputeInliner::ReplaceInlinedTensor(Expr* load) {
 void ScheduleImpl::ReverseComputeInline(const Expr& schedule_block) {
   Expr root          = this->GetRootBlock(schedule_block);
   Expr inlined_store = CheckReverseComputeInlineValidationAndGetStore(schedule_block, root);
-
   CHECK(schedule_block.As<ir::ScheduleBlockRealize>());
   auto compute_body = schedule_block.As<ir::ScheduleBlockRealize>()->schedule_block.As<ir::ScheduleBlock>()->body;
 
@@ -1434,7 +1433,7 @@ void ScheduleImpl::ReverseComputeInline(const Expr& schedule_block) {
   // Create a plan that removes the block to be inlined
   LeafBlockRemovalPlan remove_plan(schedule_block, &inliner.src_stmt, &inliner.tgt_stmt);
   remove_plan(&root);
-  inliner(&root);
+  inliner(&inliner.src_stmt);
   inliner(&root);
 }