diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh index b9d32eff3c..26e3b0bc38 100755 --- a/ci/build_wheel.sh +++ b/ci/build_wheel.sh @@ -1,5 +1,5 @@ #!/bin/bash -# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION. +# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 set -euo pipefail @@ -58,8 +58,10 @@ if [[ ${package_name} != "libraft" ]]; then ) fi -SKBUILD_CMAKE_ARGS="-DUSE_NCCL_RUNTIME_WHEEL=ON" -export SKBUILD_CMAKE_ARGS +if [[ ${package_name} == "raft-dask" ]]; then + SKBUILD_CMAKE_ARGS="-DUSE_NCCL_RUNTIME_WHEEL=ON" + export SKBUILD_CMAKE_ARGS +fi sccache --stop-server 2>/dev/null || true diff --git a/conda/recipes/libraft/recipe.yaml b/conda/recipes/libraft/recipe.yaml index ce8e2e0a30..ab3432c2c7 100644 --- a/conda/recipes/libraft/recipe.yaml +++ b/conda/recipes/libraft/recipe.yaml @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION. +# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 schema_version: 1 @@ -192,17 +192,15 @@ outputs: - cmake ${{ cmake_version }} - ${{ compiler("c") }} host: - - ${{ pin_subpackage("libraft-headers", exact=True) }} + - ${{ pin_subpackage("libraft-headers-only", exact=True) }} - cuda-version =${{ cuda_version }} - - cuda-profiler-api - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev run: - - ${{ pin_subpackage("libraft-headers", exact=True) }} + - ${{ pin_subpackage("libraft-headers-only", exact=True) }} - ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }} - - cuda-profiler-api - libcublas - libcurand - libcusolver @@ -242,17 +240,15 @@ outputs: - cmake ${{ cmake_version }} - ${{ compiler("c") }} host: - - ${{ pin_subpackage("libraft-headers", exact=True) }} + - ${{ pin_subpackage("libraft-headers-only", exact=True) }} - cuda-version =${{ cuda_version }} - - cuda-profiler-api - libcublas-dev - libcurand-dev - libcusolver-dev - libcusparse-dev run: - - ${{ pin_subpackage("libraft-headers", exact=True) }} + - ${{ pin_subpackage("libraft-headers-only", exact=True) }} - ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }} - - cuda-profiler-api - libcublas - libcurand - libcusolver diff --git a/dependencies.yaml b/dependencies.yaml index 459242e20c..0a56291454 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -82,7 +82,6 @@ files: - build_common - depends_on_librmm - depends_on_rapids_logger - - depends_on_nccl py_run_libraft: output: pyproject pyproject_dir: python/libraft @@ -92,7 +91,6 @@ files: - cuda_wheels - depends_on_librmm - depends_on_rapids_logger - - depends_on_nccl py_build_pylibraft: output: pyproject pyproject_dir: python/pylibraft