Revert "Disable the host numa virtual memory allocator for now. (dmlc…

…#10934)"

This reverts commit f4d94a1.

src/common/device_helpers.cu

@@ -7,11 +7,6 @@
 
 namespace dh {
 PinnedMemory::PinnedMemory() {
-  // Use the `GrowOnlyPinnedMemoryImpl` as the only option for now.
-  // See https://github.com/dmlc/xgboost/issues/10933
-  this->impl_.emplace<detail::GrowOnlyPinnedMemoryImpl>();
-  return;
-
 #if defined(xgboost_IS_WIN)
   this->impl_.emplace<detail::GrowOnlyPinnedMemoryImpl>();
 #else

tests/cpp/common/test_device_vector.cu

@@ -31,9 +31,6 @@ class TestVirtualMem : public ::testing::TestWithParam<CUmemLocationType> {
  public:
   void Run() {
     auto type = this->GetParam();
-    if (type == CU_MEM_LOCATION_TYPE_HOST_NUMA) {
-      GTEST_SKIP_("Host numa might require special system capabilities, skipping for now.");
-    }
     detail::GrowOnlyVirtualMemVec vec{type};
     auto prop = xgboost::cudr::MakeAllocProp(type);
     auto gran = xgboost::cudr::GetAllocGranularity(&prop);
@@ -113,6 +110,14 @@ TEST(TestVirtualMem, Version) {
   xgboost::curt::DrVersion(&major, &minor);
   LOG(INFO) << "Latest supported CUDA version by the driver:" << major << "." << minor;
   PinnedMemory pinned;
+#if defined(xgboost_IS_WIN)
   ASSERT_FALSE(pinned.IsVm());
+#else  // defined(xgboost_IS_WIN)
+  if (major >= 12 && minor >= 5) {
+    ASSERT_TRUE(pinned.IsVm());
+  } else {
+    ASSERT_FALSE(pinned.IsVm());
+  }
+#endif  // defined(xgboost_IS_WIN)
 }
 }  // namespace dh