Remove all use of DeviceQuantileDMatrix. (#8665)

2023-01-17 00:04:10 +08:00
parent 0ae8df9a65
commit d6018eb4b9
10 changed files with 57 additions and 53 deletions
--- a/tests/python-gpu/test_large_input.py
+++ b/tests/python-gpu/test_large_input.py
@@ -17,7 +17,7 @@ def test_large_input():
    assert (np.log2(m * n) > 31)
    X = cp.ones((m, n), dtype=np.float32)
    y = cp.ones(m)
-    dmat = xgb.DeviceQuantileDMatrix(X, y)
+    dmat = xgb.QuantileDMatrix(X, y)
    booster = xgb.train({"tree_method": "gpu_hist", "max_depth": 1}, dmat, 1)
    del y
    booster.inplace_predict(X)