Skip to content

Commit

Permalink
[distributed] Remove unused variable in test_composability/test_pp_co…
Browse files Browse the repository at this point in the history
…mposability.py (pytorch#143191)

Pull Request resolved: pytorch#143191
Approved by: https://github.com/mori360
  • Loading branch information
rec authored and pytorchmergebot committed Dec 14, 2024
1 parent de48413 commit 91bf2e1
Showing 1 changed file with 0 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -537,12 +537,6 @@ def apply_tp(
else:
pipeline_schedule.step()

# accumulate losses across pipeline microbatches
loss = (
torch.mean(torch.stack(losses))
if is_last_stage
else torch.Tensor([-1.0])
)
for optimizer in optimizers:
optimizer.step()

Expand Down

0 comments on commit 91bf2e1

Please sign in to comment.