dask / dask/distributed

[Fea] Run function on all workers within an annotate context

Open
#4,608 4 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.7k
Forks
778
Avg merge
2h 50m
Merged PRs (30d)
3

Description

`client.run` allows users to pass a list of workers to run a function on specific workers. It would be good if `client.run` could be used in conjunction with the annotate context to run on all workers with a specific resource tag.

Eg:
```python
with dask.annotate(resources={'GPU': 1}):
client.run(gpu_init_func)
```

cc: @jakirkham @beckernick

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.