Skip to content

Commit

Permalink
Silence KINETO
Browse files Browse the repository at this point in the history
  • Loading branch information
carmocca committed Apr 4, 2024
1 parent 2480b1c commit 1464fe1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion thunder/tests/distributed/test_ddp.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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)),
Expand Down

0 comments on commit 1464fe1

Please sign in to comment.