From fae5b736d6c11d6c70a0993ae5789f093f36806b Mon Sep 17 00:00:00 2001 From: Mark Harris <783069+harrism@users.noreply.github.com> Date: Mon, 8 Jan 2024 17:22:28 -0800 Subject: [PATCH] Add detail::available_device_memory back as an alias of rmm::available_device_memory --- include/rmm/cuda_device.hpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/include/rmm/cuda_device.hpp b/include/rmm/cuda_device.hpp index 73c49cd63..f1976b4c7 100644 --- a/include/rmm/cuda_device.hpp +++ b/include/rmm/cuda_device.hpp @@ -116,6 +116,12 @@ inline std::pair available_device_memory() return {free, total}; } +// TODO: temporary alias for backward compatibility. Remove once dependent libraries like cuGraph +// and cuDF are fixed to not use the old `rmm::defail::available_device_memory` function. +namespace detail { +const auto available_device_memory = rmm::available_device_memory; +} + /** * @brief Returns the approximate specified percent of free device memory on the current CUDA * device, aligned to the nearest CUDA allocation size.