Resolve dask performance issues (#4914)

* Set dask client.map as impure function * Remove nrows * Remove slow check in verbose mode
2019-10-10 16:01:30 +13:00
parent 80977182c5
commit aefb1e5c2f
2 changed files with 3 additions and 9 deletions
--- a/src/common/device_helpers.cuh
+++ b/src/common/device_helpers.cuh
@@ -251,7 +251,6 @@ public:
    int current_device;
    safe_cuda(cudaGetDevice(&current_device));
    stats_.RegisterAllocation(ptr, n);
-    CHECK_LE(stats_.peak_allocated_bytes, dh::TotalMemory(current_device));
  }
  void RegisterDeallocation(void *ptr, size_t n) {
    if (!xgboost::ConsoleLogger::ShouldLog(xgboost::ConsoleLogger::LV::kDebug))