Skip to content

Commit

Permalink
Avoid creating CUDA context in LocalCUDACluster parent process (#765)
Browse files Browse the repository at this point in the history
This is helpful for preventing an extra, unnecessary, CUDA context. However, this is only helpful if `Client` is not created in the same process as `LocalCUDACluster`, as otherwise it will anyway create a CUDA context.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - Mads R. B. Kristensen (https://github.com/madsbk)

URL: #765
  • Loading branch information
pentschev authored Oct 28, 2021
1 parent ee58ad5 commit 1d547a0
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions dask_cuda/local_cuda_cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -327,6 +327,7 @@ def __init__(
self.host = kwargs.get("host", None)

initialize(
create_cuda_context=False,
enable_tcp_over_ucx=enable_tcp_over_ucx,
enable_nvlink=enable_nvlink,
enable_infiniband=enable_infiniband,
Expand Down

0 comments on commit 1d547a0

Please sign in to comment.