Searched defs:timing_enabled (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_nccl.py | 3604 def started_or_scheduled(self, timing_enabled): argument
3609 def _verify_trace(self, t, include_collectives, timing_enabled, is_json): argument
3668 def test_short_json(self, timing_enabled, include_collectives): argument
3694 def test_short_pickle(self, timing_enabled, include_collectives): argument
3822 def test_trace_while_active(self, timing_enabled, only_active): argument
3873 def test_trace_while_stuck(self, timing_enabled): argument
3940 def test_batched_send_recv(self, op_sizes_per_coalesce, timing_enabled): argument
4042 def test_individual_send_recv(self, op_sizes, timing_enabled): argument
4098 def test_coalescing_manager_collective(self, timing_enabled): argument
4203 def test_timeout_dumps(self, timing_enabled): argument
|