search

rapidsai / rapids-examples

33 stars 24 forks source link

[REVIEW]Use cupy RawKernel and Numba to get top k values #45

Closed VibhuJawa closed 2 years ago

VibhuJawa commented 2 years ago

Pushed this PR to use cupy RawKernel and Numba to get top k values.

This speeds it up by a factor of 10x.

review-notebook-app[bot] commented 2 years ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB