From 5884680a66b3a6d0dc6917ee4780a43c5692dd1d Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Thu, 26 Sep 2024 14:43:00 +0800
Subject: [PATCH] Fix.

---
 src/common/hist_util.cuh | 2 +-
 src/common/quantile.cuh  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/common/hist_util.cuh b/src/common/hist_util.cuh
index 47506805353b..66463ef2fe86 100644
--- a/src/common/hist_util.cuh
+++ b/src/common/hist_util.cuh
@@ -337,7 +337,7 @@ void ProcessWeightedSlidingWindow(Context const* ctx, Batch batch, MetaInfo cons
                                   int num_cuts_per_feature, bool is_ranking, float missing,
                                   size_t columns, size_t begin, size_t end,
                                   SketchContainer* sketch_container) {
-  SetDevice(ctx->Ordinal());
+  curt::SetDevice(ctx->Ordinal());
   info.weights_.SetDevice(ctx->Device());
   auto weights = info.weights_.ConstDeviceSpan();
 
diff --git a/src/common/quantile.cuh b/src/common/quantile.cuh
index 4d849540af9f..1b60670d0d68 100644
--- a/src/common/quantile.cuh
+++ b/src/common/quantile.cuh
@@ -206,7 +206,7 @@ class SketchContainer {
   template <typename KeyComp = thrust::equal_to<size_t>>
   size_t Unique(Context const* ctx, KeyComp key_comp = thrust::equal_to<size_t>{}) {
     timer_.Start(__func__);
-    SetDevice(ctx->Ordinal());
+    curt::SetDevice(ctx->Ordinal());
     this->columns_ptr_.SetDevice(ctx->Device());
     Span<OffsetT> d_column_scan = this->columns_ptr_.DeviceSpan();
     CHECK_EQ(d_column_scan.size(), num_columns_ + 1);