"vscode:/vscode.git/clone" did not exist on "217937221b6845913502371aba554a3357fbccfb"
Unverified Commit 084a01fd authored by Anthony Platanios's avatar Anthony Platanios Committed by GitHub
Browse files

[Bugfix] [Easy] Fixed a bug in the multiprocessing GPU executor. (#6770)

parent 062a1d0f
import asyncio
import os
import signal
import threading
import weakref
from functools import partial
from typing import Any, List, Optional
......@@ -115,8 +116,9 @@ class MultiprocessingGPUExecutor(DistributedGPUExecutor):
if executor := ref():
executor.shutdown()
signal.signal(signal.SIGINT, shutdown)
signal.signal(signal.SIGTERM, shutdown)
if threading.current_thread() is threading.main_thread():
signal.signal(signal.SIGINT, shutdown)
signal.signal(signal.SIGTERM, shutdown)
self.driver_worker = self._create_worker(
distributed_init_method=distributed_init_method)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment