diff --git a/test/distributed/_composable/test_composability/test_pp_composability.py b/test/distributed/_composable/test_composability/test_pp_composability.py index c82b08bb21f01..0239ef05242fa 100644 --- a/test/distributed/_composable/test_composability/test_pp_composability.py +++ b/test/distributed/_composable/test_composability/test_pp_composability.py @@ -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()