diff --git a/thunder/tests/distributed/test_ddp.py b/thunder/tests/distributed/test_ddp.py index a32a03160b..1d549ee6da 100644 --- a/thunder/tests/distributed/test_ddp.py +++ b/thunder/tests/distributed/test_ddp.py @@ -19,7 +19,6 @@ from torch.testing import assert_close, make_tensor import thunder -from thunder import INTERPRETATION_OPTIONS import thunder.torch as ltorch from thunder.core import devices from thunder.distributed import FSDPBucketingStrategy, FSDPType @@ -510,6 +509,7 @@ def test_rematerialize_all_gather(self): self.assertTrue(all(bsym.sym.id != PrimIDs.ALL_GATHER for bsym in bwd_trc.bound_symbols)) self.assertTrue(any(bsym.sym.id == PrimIDs.ALL_GATHER for bsym in result_bwd_trc.bound_symbols)) + @unittest.mock.patch.dict(os.environ, {"KINETO_LOG_LEVEL": "5"}) # silence torch.profiler logs @common_utils.parametrize( "executor,bucket_size_in_mb,dataset_size", product(tuple(executors_map.keys()), (0, 25), (1, 2)),