From dc68abb9f68c36ff6d018161634d09745e2380e9 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Mon, 3 Jun 2024 11:52:45 -0700 Subject: [PATCH] move kvikio around --- dependencies.yaml | 10 ++++++++-- pyproject.toml | 1 + 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/dependencies.yaml b/dependencies.yaml index c8b29272..a66c5782 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -160,15 +160,15 @@ dependencies: common: - output_types: [conda, requirements, pyproject] packages: - - kvikio==24.8.*,>=0.0.0a0 - pytest - pytest-cov - output_types: [conda] packages: - &cudf_conda cudf==24.8.*,>=0.0.0a0 - &dask_cudf_conda dask-cudf==24.8.*,>=0.0.0a0 + - &distributed_ucxx_conda distributed-ucxx==0.39.*,>=0.0.0a0 + - &kvikio_conda kvikio==24.8.*,>=0.0.0a0 - &ucx_py_conda ucx-py==0.39.*,>=0.0.0a0 - - distributed-ucxx==0.39.*,>=0.0.0a0 - ucx-proc=*=gpu - ucxx==0.39.*,>=0.0.0a0 specific: @@ -188,14 +188,20 @@ dependencies: packages: - cudf-cu12==24.8.*,>=0.0.0a0 - dask-cudf-cu12==24.8.*,>=0.0.0a0 + - distributed-ucxx-cu12==24.8.*,>=0.0.0a0 + - kvikio-cu12==24.8.*,>=0.0.0a0 - ucx-py-cu12==0.39.*,>=0.0.0a0 - matrix: {cuda: "11.*"} packages: - cudf-cu11==24.8.*,>=0.0.0a0 - dask-cudf-cu11==24.8.*,>=0.0.0a0 + - distributed-ucxx-cu12==24.8.*,>=0.0.0a0 + - kvikio-cu12==24.8.*,>=0.0.0a0 - ucx-py-cu11==0.39.*,>=0.0.0a0 - matrix: packages: - *cudf_conda - *dask_cudf_conda + - *distributed_ucxx_conda + - *kvikio_conda - *ucx_py_conda diff --git a/pyproject.toml b/pyproject.toml index 77fcc3c7..235429a7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -52,6 +52,7 @@ docs = [ test = [ "cudf==24.8.*,>=0.0.0a0", "dask-cudf==24.8.*,>=0.0.0a0", + "distributed-ucxx==0.39.*,>=0.0.0a0", "kvikio==24.8.*,>=0.0.0a0", "pytest", "pytest-cov",