[EM] Support CPU quantile objective for external memory. (#10751)

This commit is contained in:
Jiaming Yuan
2024-08-27 04:16:57 +08:00
committed by GitHub
parent 12c6b7ceea
commit d6ebcfb032
13 changed files with 163 additions and 36 deletions

View File

@@ -4,6 +4,7 @@ import pytest
from hypothesis import given, settings, strategies
from xgboost.testing import no_cupy
from xgboost.testing.updater import check_quantile_loss_extmem
sys.path.append("tests/python")
from test_data_iterator import run_data_iterator
@@ -56,3 +57,8 @@ def test_cpu_data_iterator() -> None:
use_cupy=True,
on_host=False,
)
def test_quantile_objective() -> None:
with pytest.raises(ValueError, match="external memory"):
check_quantile_loss_extmem(2, 2, 2, "hist", "cuda")