MCPcopy Create free account
hub / github.com/dmlc/xgboost / test_cpu_data_iterator

Function test_cpu_data_iterator

tests/python-gpu/test_gpu_data_iterator.py:60–71  ·  view source on GitHub ↗

Make sure CPU algorithm can handle GPU inputs

()

Source from the content-addressed store, hash-verified

58
59
60def test_cpu_data_iterator() -> None:
61 """Make sure CPU algorithm can handle GPU inputs"""
62 run_data_iterator(
63 1024,
64 2,
65 3,
66 "approx",
67 device="cuda",
68 subsample=False,
69 use_cupy=True,
70 on_host=False,
71 )
72
73
74@given(

Callers

nothing calls this directly

Calls 1

run_data_iteratorFunction · 0.90

Tested by

no test coverage detected