Skip to content

Use static cudart #112

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
rapids-bot merged 3 commits into from
May 12, 2026
Merged

Use static cudart #112

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
d35f3fd
Use static cudart
KyleFromNVIDIA May 12, 2026
3c13543
INTERFACE link cudart_static
KyleFromNVIDIA May 12, 2026
7fd415f
Unconditionally link cudart_static
KyleFromNVIDIA May 12, 2026
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
2 changes: 0 additions & 2 deletions conda/recipes/libnvforest/recipe.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,6 @@ outputs:
- cuda-cudart-dev
run:
- ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
- cuda-cudart
- librmm =${{ minor_version }}
- rapids-logger =0.2
- treelite ${{ treelite_version }}
Expand Down Expand Up @@ -139,7 +138,6 @@ outputs:
run:
- ${{ pin_compatible("cuda-version", upper_bound="x", lower_bound="x") }}
- ${{ pin_subpackage("libnvforest", exact=True) }}
- cuda-cudart
ignore_run_exports:
by_name:
- cuda-cudart
Expand Down
12 changes: 3 additions & 9 deletions cpp/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,6 @@ option(DISABLE_DEPRECATION_WARNINGS "Disable deprecation warnings " ON)
option(DISABLE_OPENMP "Disable OpenMP" OFF)
option(NVTX "Enable nvtx markers" OFF)
option(USE_CCACHE "Cache build artifacts with ccache" OFF)
option(CUDA_STATIC_RUNTIME "Statically link the CUDA runtime" OFF)
option(NVFOREST_USE_RAFT_STATIC "Build and statically link the RAFT library" OFF)
option(NVFOREST_USE_TREELITE_STATIC "Build and statically link the treelite library" OFF)
option(NVFOREST_EXPORT_TREELITE_LINKAGE "Whether to publicly or privately link treelite to libnvforest++" OFF)
Expand All @@ -79,7 +78,6 @@ message(VERBOSE "NVFOREST_CPP: Enabling kernelinfo in nvcc: ${CUDA_ENABLE_KERNEL
message(VERBOSE "NVFOREST_CPP: Enabling lineinfo in nvcc: ${CUDA_ENABLE_LINE_INFO}")
message(VERBOSE "NVFOREST_CPP: Enabling nvtx markers: ${NVTX}")
message(VERBOSE "NVFOREST_CPP: Cache build artifacts with ccache: ${USE_CCACHE}")
message(VERBOSE "NVFOREST_CPP: Statically link the CUDA runtime: ${CUDA_STATIC_RUNTIME}")
message(VERBOSE "NVFOREST_CPP: Build and statically link RAFT library: ${NVFOREST_USE_RAFT_STATIC}")
message(VERBOSE "NVFOREST_CPP: Build and statically link Treelite library: ${NVFOREST_USE_TREELITE_STATIC}")

Expand Down Expand Up @@ -150,7 +148,7 @@ if(NOT DISABLE_OPENMP)
endif()

# CUDA runtime
rapids_cuda_init_runtime(USE_STATIC ${CUDA_STATIC_RUNTIME})
rapids_cuda_init_runtime(USE_STATIC ON)

# * find CUDAToolkit package
# * determine GPU architectures
Expand Down Expand Up @@ -296,12 +294,8 @@ endif()
# These are always private:
list(APPEND _nvforest_cpp_private_libs ${TREELITE_LIBS} ${OpenMP_CXX_LIB_NAMES})

set(_nvforest_cpp_libs_var_name "_nvforest_cpp_public_libs")
if(CUDA_STATIC_RUNTIME)
set(_nvforest_cpp_libs_var_name "_nvforest_cpp_private_libs")
# Add CTK include paths because we're going to make our CTK library links private below
target_include_directories(${NVFOREST_CPP_TARGET} SYSTEM PUBLIC ${CUDAToolkit_INCLUDE_DIRS})
endif()
# Add CTK include paths because we're going to make our CTK library links private below
target_include_directories(${NVFOREST_CPP_TARGET} SYSTEM PUBLIC ${CUDAToolkit_INCLUDE_DIRS})

target_link_libraries(
${NVFOREST_CPP_TARGET}
Expand Down
4 changes: 0 additions & 4 deletions docs/source/build.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -292,10 +292,6 @@ nvForest's cmake has the following configurable flags available:
- [ON, OFF]
- OFF
- Whether to cache build artifacts with ccache.
* - CUDA_STATIC_RUNTIME
- [ON, OFF]
- OFF
- Whether to statically link the CUDA runtime.
* - NVFOREST_USE_RAFT_STATIC
- [ON, OFF]
- OFF
Expand Down
3 changes: 0 additions & 3 deletions python/libnvforest/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,9 +42,6 @@ set(NVFOREST_PYTHON_TREELITE_TARGET treelite::treelite_static)
set(NVFOREST_USE_TREELITE_STATIC ON)
set(NVFOREST_EXCLUDE_TREELITE_FROM_ALL ON)

# --- CUDA --- #
set(CUDA_STATIC_RUNTIME ON)

add_subdirectory(../../cpp nvforest-cpp)

set_property(
Expand Down
Loading