File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff 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_;
Original file line number Diff line number Diff 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 }
You can’t perform that action at this time.
0 commit comments