Home
last modified time | relevance | path

Searched refs:MultiprocessingRequestQueue (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/elastic/timer/
H A Dlocal_timer_test.py16 from torch.distributed.elastic.timer.local_timer import MultiprocessingRequestQueue
138 request_queue = MultiprocessingRequestQueue(mp_queue)
160 request_queue = MultiprocessingRequestQueue(mp_queue)
185 request_queue = MultiprocessingRequestQueue(mp_queue)
/aosp_15_r20/external/pytorch/torch/distributed/elastic/timer/
H A Dlocal_timer.py47 class MultiprocessingRequestQueue(RequestQueue): class
90 super().__init__(MultiprocessingRequestQueue(mp_queue), max_interval, daemon)