Protect child process enumeration in AutoTVM (apache#7887)

trevor-m · May 6, 2021 · de20d24 · de20d24
1 parent 33bbae9
commit de20d24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/autotvm/measure/local_executor.py b/python/tvm/autotvm/measure/local_executor.py
@@ -37,9 +37,9 @@ def kill_child_processes(parent_pid, sig=signal.SIGTERM):
     """kill all child processes recursively"""
     try:
         parent = psutil.Process(parent_pid)
+        children = parent.children(recursive=True)
     except psutil.NoSuchProcess:
         return
-    children = parent.children(recursive=True)
     for process in children:
         try:
             process.send_signal(sig)