Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[JVM-Packages] Allow XGBoost jvm package run on GPU without spark-rapids #11184

Merged
merged 1 commit into from
Jan 26, 2025

Conversation

wbo4958
Copy link
Contributor

@wbo4958 wbo4958 commented Jan 26, 2025

When I tried to run a simple xgboost4j spark gpu example with GPU, xgboost throws exception indicating "Couldn't find ai/rapids/Table" which is in spark-rapids. Which is not correct, since xgboost4j-spark-gpu shouldn't be coupled with spark-rapids.

This PR fixes this issue.

@wbo4958 wbo4958 changed the title [JVM-Packages] Allow XGBoost jvm package run on GPU without rapids [JVM-Packages] Allow XGBoost jvm package run on GPU without spark-rapids Jan 26, 2025
@wbo4958
Copy link
Contributor Author

wbo4958 commented Jan 26, 2025

Hi @trivialfis, Please help review it. Thx.

@trivialfis trivialfis merged commit 766bfcc into dmlc:master Jan 26, 2025
58 checks passed
@wbo4958 wbo4958 deleted the without-rapids branch January 26, 2025 06:34
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants