Skip to content

GridSearchCV & RandomizedSearchCV GPU support #2406

@napetrov

Description

@napetrov

Discussed in #2350

Originally posted by idkSeth March 8, 2025
In order to offload to GPU, input data needs to be passed as dpctl.tensor.usm_ndarray as per the guide.

GridSearchCV and RandomizedSearchCV is not patched and when trying to use them with the input data as dpctl.tensor.usm_ndarray, an indexing error is raised.

Can searching be patched to allow for hyper-parameter searching on GPU/CPU with usm_ndarray?

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions