Don't estimate sketch batch size when rmm is used. (#6807)

2021-03-31 15:29:56 +08:00
parent 10ae0f9511
commit 3039dd194b
2 changed files with 9 additions and 0 deletions
--- a/tests/cpp/common/test_hist_util.cu
+++ b/tests/cpp/common/test_hist_util.cu
@@ -45,6 +45,10 @@ TEST(HistUtil, DeviceSketch) {
 }

 TEST(HistUtil, SketchBatchNumElements) {
+#if defined(XGBOOST_USE_RMM) && XGBOOST_USE_RMM == 1
+  LOG(WARNING) << "Test not runnable with RMM enabled.";
+  return;
+#endif  // defined(XGBOOST_USE_RMM) && XGBOOST_USE_RMM == 1
  size_t constexpr kCols = 10000;
  int device;
  dh::safe_cuda(cudaGetDevice(&device));