Fix for new numbapro
gpu k-select (WIP)