From 9eaeab518370357a1ea22a2d5cd845dbd6a76a22 Mon Sep 17 00:00:00 2001
From: Peter Andreas Entschev <peter@entschev.com>
Date: Mon, 22 Nov 2021 02:54:20 -0800
Subject: [PATCH] Add create_cuda_context UCX config from Distributed

This is necessary to fill the missing create_cuda_context configuration
option recently added in Distributed. In
https://github.com/rapidsai/dask-cuda/pull/792 it will be used to
simplify UCX configuration.
---
 dask_cuda/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dask_cuda/utils.py b/dask_cuda/utils.py
index 4f863d5e9..15794e6eb 100644
--- a/dask_cuda/utils.py
+++ b/dask_cuda/utils.py
@@ -303,6 +303,7 @@ def get_ucx_config(
         "rdmacm": None,
         "net-devices": None,
         "cuda_copy": None,
+        "create_cuda_context": None,
         "reuse-endpoints": not _ucx_111,
     }
     if enable_tcp_over_ucx or enable_infiniband or enable_nvlink: