Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion ci/build_wheel_cudf.sh
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -27,13 +27,19 @@ export RAPIDS_PY_API

./ci/build_wheel.sh cudf ${package_dir} --stable

RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
RAPIDS_CUDA_MAJOR_MINOR="${RAPIDS_CUDA_VERSION%.*}"

# repair wheels and write to the location that artifact-uploading code expects to find them
python -m auditwheel repair \
--exclude libcudf.so \
--exclude libnvcomp.so.* \
--exclude libkvikio.so \
--exclude librapids_logger.so \
--exclude librmm.so \
--exclude libnvrtc.so.${RAPIDS_CUDA_MAJOR} \
--exclude libnvrtc-builtins.so.${RAPIDS_CUDA_MAJOR_MINOR} \
--exclude libnvJitLink.so.${RAPIDS_CUDA_MAJOR} \
-w "${RAPIDS_WHEEL_BLD_OUTPUT_DIR}" \
${package_dir}/dist/*

Expand Down
6 changes: 6 additions & 0 deletions ci/build_wheel_cudf_streaming.sh
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,9 @@ if grep -Fq "performance hint:" cudf-streaming-wheel-build-output.log; then
exit 1
fi

RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
RAPIDS_CUDA_MAJOR_MINOR="${RAPIDS_CUDA_VERSION%.*}"

# repair wheels and write to the location that artifact-uploading code expects to find them
python -m auditwheel repair \
--exclude libcudf.so \
Expand All @@ -63,6 +66,9 @@ python -m auditwheel repair \
--exclude librmm.so \
--exclude libucxx.so \
--exclude libucp.so.0 \
--exclude libnvrtc.so.${RAPIDS_CUDA_MAJOR} \
--exclude libnvrtc-builtins.so.${RAPIDS_CUDA_MAJOR_MINOR} \
--exclude libnvJitLink.so.${RAPIDS_CUDA_MAJOR} \
-w "${RAPIDS_WHEEL_BLD_OUTPUT_DIR}" \
${package_dir}/dist/*

Expand Down
6 changes: 6 additions & 0 deletions ci/build_wheel_libcudf.sh
Original file line number Diff line number Diff line change
Expand Up @@ -31,12 +31,18 @@ export PIP_NO_BUILD_ISOLATION=0
export SKBUILD_CMAKE_ARGS="-DUSE_NVCOMP_RUNTIME_WHEEL=ON"
./ci/build_wheel.sh "${package_name}" "${package_dir}"

RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
RAPIDS_CUDA_MAJOR_MINOR="${RAPIDS_CUDA_VERSION%.*}"

# repair wheels and write to the location that artifact-uploading code expects to find them
python -m auditwheel repair \
--exclude libkvikio.so \
--exclude libnvcomp.so.5 \
--exclude librapids_logger.so \
--exclude librmm.so \
--exclude libnvrtc.so.${RAPIDS_CUDA_MAJOR} \
--exclude libnvrtc-builtins.so.${RAPIDS_CUDA_MAJOR_MINOR} \
--exclude libnvJitLink.so.${RAPIDS_CUDA_MAJOR} \
-w "${RAPIDS_WHEEL_BLD_OUTPUT_DIR}" \
${package_dir}/dist/*

Expand Down
6 changes: 6 additions & 0 deletions ci/build_wheel_libcudf_streaming.sh
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,9 @@ export PIP_NO_BUILD_ISOLATION=0

./ci/build_wheel.sh "${package_name}" "${package_dir}"

RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
RAPIDS_CUDA_MAJOR_MINOR="${RAPIDS_CUDA_VERSION%.*}"

# repair wheels and write to the location that artifact-uploading code expects to find them
python -m auditwheel repair \
--exclude libcudf.so \
Expand All @@ -46,6 +49,9 @@ python -m auditwheel repair \
--exclude librmm.so \
--exclude libucxx.so \
--exclude libucp.so.0 \
--exclude libnvrtc.so.${RAPIDS_CUDA_MAJOR} \
--exclude libnvrtc-builtins.so.${RAPIDS_CUDA_MAJOR_MINOR} \
--exclude libnvJitLink.so.${RAPIDS_CUDA_MAJOR} \
-w "${RAPIDS_WHEEL_BLD_OUTPUT_DIR}" \
${package_dir}/dist/*

Expand Down
6 changes: 6 additions & 0 deletions ci/build_wheel_pylibcudf.sh
Original file line number Diff line number Diff line change
Expand Up @@ -32,13 +32,19 @@ if grep -Fq "performance hint:" pylibcudf-wheel-build-output.log; then
exit 1
fi

RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
RAPIDS_CUDA_MAJOR_MINOR="${RAPIDS_CUDA_VERSION%.*}"

# repair wheels and write to the location that artifact-uploading code expects to find them
python -m auditwheel repair \
--exclude libcudf.so \
--exclude libnvcomp.so.* \
--exclude libkvikio.so \
--exclude librapids_logger.so \
--exclude librmm.so \
--exclude libnvrtc.so.${RAPIDS_CUDA_MAJOR} \
--exclude libnvrtc-builtins.so.${RAPIDS_CUDA_MAJOR_MINOR} \
--exclude libnvJitLink.so.${RAPIDS_CUDA_MAJOR} \
-w "${RAPIDS_WHEEL_BLD_OUTPUT_DIR}" \
${package_dir}/dist/*

Expand Down
6 changes: 3 additions & 3 deletions cpp/cmake/thirdparty/get_rtcx.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -15,16 +15,16 @@ function(find_and_configure_rtcx VERSION)
rtcx ${VERSION}
GLOBAL_TARGETS rtcx::rtcx
CPM_ARGS
GIT_REPOSITORY https://github.com/rapidsai/librtcx.git
GIT_TAG efad266c1fd9de6d8486c6ba71bfa74df063eb1f
GIT_REPOSITORY https://github.com/arhag23/librtcx.git
GIT_TAG 67b5e05f5282fca9ad04d44b7dc93d75d89f1ad1
GIT_SHALLOW FALSE
EXCLUDE_FROM_ALL ${CUDF_EXCLUDE_DEPS_FROM_ALL}
)

# When CPM fetches from source (add_subdirectory), embed.cmake is not auto-included. Include it
# explicitly so add_embed/embed_includes/embed functions are available.
if(rtcx_ADDED OR DEFINED CPM_rtcx_SOURCE)
include("${rtcx_SOURCE_DIR}/embed.cmake")
include("${rtcx_SOURCE_DIR}/cmake/modules/embed.cmake")
endif()
endfunction()

Expand Down
2 changes: 1 addition & 1 deletion cpp/src/jit/cache.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

#include <cudf_cuda_embed.hpp>
#include <jit/cache.hpp>
#include <rtcx.hpp>
#include <rtcx/rtcx.hpp>
#include <runtime/context.hpp>

#define XXH_INLINE_ALL
Expand Down
2 changes: 1 addition & 1 deletion cpp/src/jit/cache.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

#include <rmm/cuda_stream_view.hpp>

#include <rtcx.hpp>
#include <rtcx/rtcx.hpp>

namespace CUDF_EXPORT cudf {

Expand Down
2 changes: 1 addition & 1 deletion cpp/src/jit/helpers.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
#include <cudf/utilities/type_dispatcher.hpp>

#include <jit/cache.hpp>
#include <rtcx.hpp>
#include <rtcx/rtcx.hpp>
#include <runtime/context.hpp>

#include <format>
Expand Down
Loading