From 91bf2e16debdc41f5dde2bb5cc8e4f39f8955d4e Mon Sep 17 00:00:00 2001 From: Tom Ritchford Date: Fri, 13 Dec 2024 12:55:25 +0000 Subject: [PATCH] [distributed] Remove unused variable in test_composability/test_pp_composability.py (#143191) Pull Request resolved: https://github.com/pytorch/pytorch/pull/143191 Approved by: https://github.com/mori360 --- .../_composable/test_composability/test_pp_composability.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/test/distributed/_composable/test_composability/test_pp_composability.py b/test/distributed/_composable/test_composability/test_pp_composability.py index c82b08bb21f..0239ef05242 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 @@ class ComposabilityTest(MultiProcessTestCase): 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()