Skip to content

Commit

Permalink
Merge branch 'branch-23.08' into cuda-120-revert-branch
Browse files Browse the repository at this point in the history
  • Loading branch information
pentschev authored Jul 19, 2023
2 parents 4f9d6bf + 616d1ae commit f4fe592
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions dask_cuda/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
import math
import operator
import os
import pickle
import time
import warnings
from contextlib import suppress
Expand Down Expand Up @@ -716,14 +717,20 @@ def get_worker_config(dask_worker):
# assume homogeneous cluster
plugin_vals = dask_worker.plugins.values()
ret = {}

# device and host memory configuration
for p in plugin_vals:
ret[f"[plugin] {type(p).__name__}"] = {
config = {
v: getattr(p, v)
for v in dir(p)
if not (v.startswith("_") or v in {"setup", "cores"})
}
# To send this back to the client the data will be serialised
# which might fail, so pre-emptively check
try:
pickle.dumps(config)
except TypeError:
config = "UNKNOWN CONFIG"
ret[f"[plugin] {type(p).__name__}"] = config

for mem in [
"memory_limit",
Expand Down

0 comments on commit f4fe592

Please sign in to comment.