• Nikhila Ravi's avatar
    making sorting for K >1 optional in KNN points function · 806ca361
    Nikhila Ravi authored
    Summary: Added `sorted` argument to the `knn_points` function. This came up during the benchmarking against Faiss - sorting added extra memory usage. Match the memory usage of Faiss by making sorting optional.
    
    Reviewed By: bottler, gkioxari
    
    Differential Revision: D22329070
    
    fbshipit-source-id: 0828ff9b48eefce99ce1f60089389f6885d03139
    806ca361
test_knn.py 8.81 KB