From 5b194a6e5d32fe38a9a2c419abceb4f25134820d Mon Sep 17 00:00:00 2001
From: Nader Al Awar <naderalawar@gmail.com>
Date: Thu, 30 Jan 2025 02:32:13 +0000
Subject: [PATCH] Use `_CCCL_HOST_DEVICE` for RTC

---
 cub/cub/device/dispatch/kernels/merge_sort.cuh | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/cub/cub/device/dispatch/kernels/merge_sort.cuh b/cub/cub/device/dispatch/kernels/merge_sort.cuh
index 3e20b304d11..c9a8a61395a 100644
--- a/cub/cub/device/dispatch/kernels/merge_sort.cuh
+++ b/cub/cub/device/dispatch/kernels/merge_sort.cuh
@@ -117,24 +117,24 @@ public:
     ::cuda::std::_If<uses_fallback_policy, fallback_block_sort_agent_t, default_block_sort_agent_t>;
   using merge_agent_t = ::cuda::std::_If<uses_fallback_policy, fallback_merge_agent_t, default_merge_agent_t>;
 
-  CUB_RUNTIME_FUNCTION static constexpr std::size_t block_sort_vsmem_per_block()
+  _CCCL_HOST_DEVICE static constexpr ::cuda::std::size_t block_sort_vsmem_per_block()
   {
     return detail::vsmem_helper_impl<block_sort_agent_t>::vsmem_per_block;
   }
 
-  CUB_RUNTIME_FUNCTION static constexpr std::size_t merge_vsmem_per_block()
+  _CCCL_HOST_DEVICE static constexpr ::cuda::std::size_t merge_vsmem_per_block()
   {
     return detail::vsmem_helper_impl<merge_agent_t>::vsmem_per_block;
   }
 
   template <typename PolicyT>
-  CUB_RUNTIME_FUNCTION static constexpr int BlockThreads(PolicyT /*policy*/)
+  _CCCL_HOST_DEVICE static constexpr int BlockThreads(PolicyT /*policy*/)
   {
     return policy_t::BLOCK_THREADS;
   }
 
   template <typename PolicyT>
-  CUB_RUNTIME_FUNCTION static constexpr int ItemsPerTile(PolicyT /*policy*/)
+  _CCCL_HOST_DEVICE static constexpr int ItemsPerTile(PolicyT /*policy*/)
   {
     return policy_t::ITEMS_PER_TILE;
   }