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.
CUDA>=9.0
pytorch
jby1993/gpu_batch_knn
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|