gpu_batch_knn

This repository includes a batch computation extension implementation of GPU-SME-kNN and supplies a pytorch interface. Note that all of the code can be only used for research purposes.