[back port] Don't estimate sketch batch size when rmm is used. (#6807) (#6830)

2021-04-07 00:16:39 +08:00
parent 604ae01b7a
commit d231e7c35f
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));