Reduce large allocation tests to 80% on WSL

Author: bdice

cpp/tests/mr/arena_mr_tests.cpp

@@ -4,6 +4,7 @@
  */
 
 #include "../byte_literals.hpp"
+#include "test_utils.hpp"
 
 #include <rmm/aligned.hpp>
 #include <rmm/cuda_device.hpp>
@@ -483,11 +484,12 @@ TEST_F(ArenaTest, SizeSmallerThanSuperblockSize)  // NOLINT
   EXPECT_THROW(construct_small(), rmm::logic_error);
 }
 
-TEST_F(ArenaTest, AllocateNinetyPercent)  // NOLINT
+TEST_F(ArenaTest, AllocateMostOfFreeMemory)  // NOLINT
 {
-  EXPECT_NO_THROW([]() {  // NOLINT(cppcoreguidelines-avoid-goto)
-    auto const ninety_percent = rmm::percent_of_free_device_memory(90);
-    arena_mr mr(rmm::mr::get_current_device_resource_ref(), ninety_percent);
+  auto const percent = is_wsl() ? 80 : 90;
+  EXPECT_NO_THROW([percent]() {  // NOLINT(cppcoreguidelines-avoid-goto)
+    auto const pool_size = rmm::percent_of_free_device_memory(percent);
+    arena_mr mr(rmm::mr::get_current_device_resource_ref(), pool_size);
   }());
 }
 

cpp/tests/mr/pool_mr_tests.cpp

@@ -3,6 +3,8 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
+#include "test_utils.hpp"
+
 #include <rmm/cuda_device.hpp>
 #include <rmm/detail/error.hpp>
 #include <rmm/device_buffer.hpp>
@@ -45,15 +47,14 @@ TEST(PoolTest, ThrowMaxLessThanInitial)
   EXPECT_THROW(max_less_than_initial(), rmm::logic_error);
 }
 
-TEST(PoolTest, AllocateNinetyPercent)
+TEST(PoolTest, AllocateMostOfFreeMemory)
 {
-  auto allocate_ninety = []() {
-    auto const [free, total] = rmm::available_device_memory();
-    (void)total;
-    auto const ninety_percent_pool = rmm::percent_of_free_device_memory(90);
-    pool_mr mr{rmm::mr::get_current_device_resource_ref(), ninety_percent_pool};
+  auto const percent = is_wsl() ? 80 : 90;
+  auto allocate      = [percent]() {
+    auto const pool_size = rmm::percent_of_free_device_memory(percent);
+    pool_mr mr{rmm::mr::get_current_device_resource_ref(), pool_size};
   };
-  EXPECT_NO_THROW(allocate_ninety());
+  EXPECT_NO_THROW(allocate());
 }
 
 TEST(PoolTest, TwoLargeBuffers)

cpp/tests/mr/test_utils.hpp

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION.
+ * SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION.
  * SPDX-License-Identifier: Apache-2.0
  */
 
@@ -10,8 +10,26 @@
 
 #include <cuda_runtime_api.h>
 
+#include <fstream>
+#include <string>
+
 namespace rmm::test {
 
+/**
+ * @brief Returns true if running under Windows Subsystem for Linux (WSL).
+ */
+inline bool is_wsl()
+{
+  std::ifstream proc_version("/proc/version");
+  if (proc_version.is_open()) {
+    std::string line;
+    std::getline(proc_version, line);
+    return line.find("microsoft") != std::string::npos ||
+           line.find("Microsoft") != std::string::npos;
+  }
+  return false;
+}
+
 /**
  * @brief Returns if a pointer points to a device memory or managed memory
  * allocation.