autogluon / autogluon/autogluon

[BUG] XGBoost Falling back to prediction using DMatrix due to mismatched devices

Open
#5,127 0 comments 1 reaction 0 assignees View on GitHub
bug: unconfirmed module: tabular Needs Triage
Dominant language
Python
Stars
10.7k
Forks
1.2k
Avg merge
21h 29m
Merged PRs (30d)
57

Description

I am doing a tabular fit with evaluation_quality and GPU enabled as per the documentation.

I am getting this error: /home/alexko/PycharmProjects/algo-py/.venv/lib/python3.11/site-packages/xgboost/core.py:158: UserWarning: [14:45:25] WARNING: /workspace/src/common/error_msg.cc:58: Falling back to prediction using DMatrix due to mismatched devices. This might lead to higher memory usage and slower performance. XGBoost is running on: cuda:0, while the input data is on: cpu.
#[36m(_ray_fit pid=271196)#[0m Potential solutions:
#[36m(_ray_fit pid=271196)#[0m - Use a data structure that matches the device ordinal in the booster.

I am not sure of any settings where the storage for XGBoost can be changed.

Contributor guide

Open the contributing guide

Research direction

The issue names no repository files, tests, or entry points. Start by reproducing the documented tabular fit with evaluation_quality and GPU enabled, then trace where the XGBoost booster and input data receive their devices. Done means identifying and documenting or fixing the mismatched-device path without the fallback warning.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning, performance
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.