!12928 Prevents child processes from entering sleep stat

From: @ouwenchang
Reviewed-by: 
Signed-off-by:
pull/12928/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 363e574ff8

@ -14,7 +14,6 @@
# ============================================================================ # ============================================================================
"""Write events to disk in a base directory.""" """Write events to disk in a base directory."""
import os import os
import sys
import time import time
import signal import signal
from collections import deque from collections import deque
@ -101,7 +100,9 @@ class WriterPool(ctx.Process):
with ctx.Pool(min(ctx.cpu_count(), 32)) as pool: with ctx.Pool(min(ctx.cpu_count(), 32)) as pool:
deq = deque() deq = deque()
while True: while True:
self._check_heartbeat() if self._check_heartbeat():
self._close()
return
while deq and deq[0].ready(): while deq and deq[0].ready():
for plugin, data in deq.popleft().get(): for plugin, data in deq.popleft().get():
@ -195,6 +196,4 @@ class WriterPool(ctx.Process):
"so SummaryRecord will not record data.") "so SummaryRecord will not record data.")
is_exit = True is_exit = True
if is_exit: return is_exit
self._close()
sys.exit(1)

Loading…
Cancel
Save