Skip to content

Commit 2d58512

Browse files
authored
[SOT] Fix bug of cuda_graph mode in sot (#74749)
1 parent d1c8a43 commit 2d58512

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/eager/to_static/run_program_impl.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -573,7 +573,7 @@ std::vector<paddle::Tensor> RunProgramImpl(
573573
#endif
574574

575575
auto passed_kernel_program = paddle::framework::ApplyIrPass(
576-
forward_program.get(), place, no_need_buffer_name_set);
576+
program.get(), place, no_need_buffer_name_set);
577577
const auto &new_block = passed_kernel_program->block();
578578
passed_kernel_program = paddle::framework::ApplyRemoveShadowFeedPass(
579579
std::move(passed_kernel_program), new_block, place, global_inner_scope);

0 commit comments

Comments
 (0)