A modification version of KNN_CUDA to support GPU specify - View it on GitHub