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.

Sign up for GitHub

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

Jump to bottom

[skl] Return GPU output if input is on GPU. #8950

Open

trivialfis wants to merge 1 commit into dmlc:master from trivialfis:skl-gpu-output

Member

trivialfis commented Mar 20, 2023 •

edited

To properly support GPU inputs, we still need a GPU-based score function for estimators.

todos

Address this in the C level for the predict function, use the Json config for passing in parameter.

trivialfis force-pushed the skl-gpu-output branch from 2b64f1b to c2d7259 Compare

August 22, 2023 18:06


          [skl] Return GPU output if input is on GPU.

e4df01d

pylint.

pylint.

trivialfis force-pushed the skl-gpu-output branch from c2d7259 to e4df01d Compare

October 9, 2023 07:42

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment