Open
Description
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
?