diff --git a/cpp/cmake/thirdparty/get_rmm.cmake b/cpp/cmake/thirdparty/get_rmm.cmake index 854bd3d1149..0a5263182ca 100644 --- a/cpp/cmake/thirdparty/get_rmm.cmake +++ b/cpp/cmake/thirdparty/get_rmm.cmake @@ -1,5 +1,5 @@ # ============================================================================= -# Copyright (c) 2020-2021, NVIDIA CORPORATION. +# Copyright (c) 2020-2024, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except # in compliance with the License. You may obtain a copy of the License at @@ -15,6 +15,10 @@ # This function finds rmm and sets any additional necessary environment variables. function(find_and_configure_rmm) include(${rapids-cmake-dir}/cpm/rmm.cmake) + include(${rapids-cmake-dir}/cpm/package_override.cmake) + + set(cudf_patch_dir "${CMAKE_CURRENT_FUNCTION_LIST_DIR}/patches") + rapids_cpm_package_override("${cudf_patch_dir}/rmm_override.json") # Find or install RMM rapids_cpm_rmm(BUILD_EXPORT_SET cudf-exports INSTALL_EXPORT_SET cudf-exports) diff --git a/cpp/cmake/thirdparty/patches/rmm_override.json b/cpp/cmake/thirdparty/patches/rmm_override.json new file mode 100644 index 00000000000..ce31e09994d --- /dev/null +++ b/cpp/cmake/thirdparty/patches/rmm_override.json @@ -0,0 +1,9 @@ +{ + "packages": { + "rmm": { + "version": "25.02", + "git_url": "https://github.com/vyasr/rmm.git", + "git_tag": "fix/logger_issues" + } + } +}