Enable tests with mixed devices.
(DMatrixT: Type, device: Device, tdevice: Device)
| 320 | |
| 321 | |
| 322 | def run_basic_predict(DMatrixT: Type, device: Device, tdevice: Device) -> None: |
| 323 | """Enable tests with mixed devices.""" |
| 324 | enc, reenc, encoded, y = _basic_example(device) |
| 325 | |
| 326 | Xy = DMatrixT(enc, y, enable_categorical=True) |
| 327 | booster = train({"device": tdevice}, Xy, num_boost_round=4) |
| 328 | |
| 329 | predt0 = booster.inplace_predict(reenc) |
| 330 | predt1 = booster.inplace_predict(encoded) |
| 331 | assert_allclose(device, predt0, predt1) |
| 332 | |
| 333 | fmat = DMatrixT(reenc, enable_categorical=True) |
| 334 | predt2 = booster.predict(fmat) |
| 335 | assert_allclose(device, predt0, predt2) |
| 336 | |
| 337 | |
| 338 | def run_cat_predict(device: Device) -> None: |