diff --git a/dask_cuda/cli.py b/dask_cuda/cli.py index 8d335ec3..d454129a 100644 --- a/dask_cuda/cli.py +++ b/dask_cuda/cli.py @@ -166,11 +166,13 @@ def cuda(): ) @click.option( "--set-rmm-allocator-for-libs", + "rmm_allocator_external_lib_list", default=None, show_default=True, help=""" Set RMM as the allocator for external libraries. Provide a comma-separated - list of libraries to set, e.g., "torch,cupy". Supported options are: torch, cupy.""", + list of libraries to set, e.g., "torch,cupy". + Supported options are: torch, cupy.""", ) @click.option( "--rmm-release-threshold",