Define the new device parameter. (#9362)

2023-07-13 19:30:25 +08:00
parent 2d0cd2817e
commit 04aff3af8e
63 changed files with 827 additions and 477 deletions
--- a/src/data/proxy_dmatrix.cc
+++ b/src/data/proxy_dmatrix.cc
@@ -54,6 +54,7 @@ std::shared_ptr<DMatrix> CreateDMatrixFromProxy(Context const *ctx,
    p_fmat = cuda_impl::CreateDMatrixFromProxy(ctx, proxy, missing);
  }

+  CHECK(p_fmat) << "Failed to fallback.";
  return p_fmat;
 }
 }  // namespace xgboost::data