FEA extending top-k to multi row inputs

soda-inria / sklearn-numba-dpex

Experimental plugin for scikit-learn to be able to run (some estimators) on Intel GPUs via numba-dpex.

BSD 3-Clause "New" or "Revised" License

15 stars 4 forks source link

Closed fcharras closed 1 year ago

fcharras commented 1 year ago

Extending top-k to multirow top-k, where top-k search in all rows of the input is fused in the same kernels.

fcharras commented 1 year ago

TY for review :heart:

Answered comments and/or applied suggestions in last commit