From 344ddeb9ca8fbf5e546250c3e4c52c012a977ee3 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sat, 20 Jul 2024 06:14:56 +0800 Subject: [PATCH] Drop support for CUDA legacy stream. (#10607) --- CMakeLists.txt | 1 - cmake/Utils.cmake | 8 ++------ 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index f7cf8a6cf..034d52164 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -71,7 +71,6 @@ option(HIDE_CXX_SYMBOLS "Build shared library and hide all C++ symbols" OFF) option(KEEP_BUILD_ARTIFACTS_IN_BINARY_DIR "Output build artifacts in CMake binary dir" OFF) ## CUDA option(USE_CUDA "Build with GPU acceleration" OFF) -option(USE_PER_THREAD_DEFAULT_STREAM "Build with per-thread default stream" ON) option(USE_NCCL "Build with NCCL to enable distributed GPU support." OFF) # This is specifically designed for PyPI binary release and should be disabled for most of the cases. option(USE_DLOPEN_NCCL "Whether to load nccl dynamically." OFF) diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake index 9563ec307..266cf29b3 100644 --- a/cmake/Utils.cmake +++ b/cmake/Utils.cmake @@ -80,12 +80,8 @@ function(xgboost_set_cuda_flags target) $<$:--expt-extended-lambda> $<$:--expt-relaxed-constexpr> $<$:-Xcompiler=${OpenMP_CXX_FLAGS}> - $<$:-Xfatbin=-compress-all>) - - if(USE_PER_THREAD_DEFAULT_STREAM) - target_compile_options(${target} PRIVATE - $<$:--default-stream per-thread>) - endif() + $<$:-Xfatbin=-compress-all> + $<$:--default-stream per-thread>) if(FORCE_COLORED_OUTPUT) if(FORCE_COLORED_OUTPUT AND (CMAKE_GENERATOR STREQUAL "Ninja") AND