Skip to content

Commit 4a806a7

Browse files
committed
fix, test=allcase
1 parent c7d2e63 commit 4a806a7

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

paddle/fluid/framework/device_worker.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -595,7 +595,7 @@ class SectionWorker : public DeviceWorker {
595595
std::vector<Scope*> microbatch_scopes_;
596596
const Scope* minibatch_scope_;
597597

598-
// skip, forward&backward send vars only used in 1F1B
598+
// skip&forward&backward vars are only used in 1F1B
599599
std::vector<std::string> skip_vars_;
600600
std::vector<std::string> forward_send_vars_;
601601
std::vector<std::string> backward_send_vars_;

paddle/fluid/framework/section_worker.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,6 @@ void SectionWorker::Run1F1B(std::unique_ptr<GarbageCollector> &gc) {
200200

201201
// delete forward send var at step<=(fw_step - 1)
202202
if (gc && !is_last_stage) {
203-
// 0, 1, 2
204203
for (int i = reserve_fw_send_step; i < fw_step; ++i) {
205204
DeleteUnusedTensors(*microbatch_scopes_[i], forward_send_vars_,
206205
gc.get());
@@ -223,7 +222,8 @@ void SectionWorker::Run1F1B(std::unique_ptr<GarbageCollector> &gc) {
223222
// NOTE(wangxi): will only execute once
224223
// delete forward send var at step=(num_microbatches_ - 1)
225224
if (reserve_fw_send_step < num_microbatches_ && !is_last_stage) {
226-
DeleteUnusedTensors(*microbatch_scopes_[i], forward_send_vars_, gc.get());
225+
DeleteUnusedTensors(*microbatch_scopes_[reserve_fw_send_step],
226+
forward_send_vars_, gc.get());
227227
++reserve_fw_send_step;
228228
}
229229
}

0 commit comments

Comments
 (0)