Remove GPU executor (#8399)

dask · Dec 14, 2023 · 61edd73 · 61edd73
1 parent f2e1e51
commit 61edd73
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 17 deletions.
diff --git a/distributed/tests/test_worker.py b/distributed/tests/test_worker.py
@@ -30,7 +30,6 @@
 from dask.system import CPU_COUNT
 from dask.utils import tmpfile
 
-import distributed
 from distributed import (
     Client,
     Event,
@@ -46,7 +45,6 @@
 from distributed.comm.utils import OFFLOAD_THRESHOLD
 from distributed.compatibility import LINUX, WINDOWS
 from distributed.core import CommClosedError, Status, rpc
-from distributed.diagnostics import nvml
 from distributed.diagnostics.plugin import ForwardOutput
 from distributed.metrics import time
 from distributed.protocol import pickle
@@ -2267,17 +2265,6 @@ async def test_process_executor_raise_exception(c, s, a, b):
             await future
 
 
-@pytest.mark.gpu
-@gen_cluster(client=True, nthreads=[("127.0.0.1", 1)])
-async def test_gpu_executor(c, s, w):
-    if nvml.device_get_count() > 0:
-        e = w.executors["gpu"]
-        assert isinstance(e, distributed.threadpoolexecutor.ThreadPoolExecutor)
-        assert e._max_workers == 1
-    else:
-        assert "gpu" not in w.executors
-
-
 async def assert_task_states_on_worker(
     expected: dict[str, str], worker: Worker
 ) -> None:

diff --git a/distributed/worker.py b/distributed/worker.py
@@ -636,10 +636,6 @@ def __init__(
             "offload": utils._offload_executor,
             "actor": ThreadPoolExecutor(1, thread_name_prefix="Dask-Actor-Threads"),
         }
-        if nvml.device_get_count() > 0:
-            self.executors["gpu"] = ThreadPoolExecutor(
-                1, thread_name_prefix="Dask-GPU-Threads"
-            )
 
         # Find the default executor
         if executor == "offload":