LICENSE
MANIFEST.in
README.md
setup.py
torchpq/CustomModule.py
torchpq/__init__.py
torchpq/_version.py
torchpq/metric.py
torchpq/util.py
torchpq.egg-info/PKG-INFO
torchpq.egg-info/SOURCES.txt
torchpq.egg-info/dependency_links.txt
torchpq.egg-info/requires.txt
torchpq.egg-info/top_level.txt
torchpq/clustering/KMeans.py
torchpq/clustering/KMeansOld.py
torchpq/clustering/MinibatchKMeans.py
torchpq/clustering/MultiKMeans.py
torchpq/clustering/MultiKMeansOld.py
torchpq/clustering/__init__.py
torchpq/codec/BaseCodec.py
torchpq/codec/PQCodec.py
torchpq/codec/SQCodec.py
torchpq/codec/VQCodec.py
torchpq/codec/__init__.py
torchpq/container/BaseContainer.py
torchpq/container/CellContainer.py
torchpq/container/DistributedCellContainer.py
torchpq/container/FlatContainer.py
torchpq/container/FlatContainerGroup.py
torchpq/container/__init__.py
torchpq/fn/IVFPQTopk.py
torchpq/fn/Topk.py
torchpq/fn/__init__.py
torchpq/index/FlatIndex.py
torchpq/index/IVFPQIndex.py
torchpq/index/IVFPQRIndex.py
torchpq/index/__init__.py
torchpq/kernels/BMMCuda.py
torchpq/kernels/ComputeCentroidsCuda.py
torchpq/kernels/ComputeProductCuda.py
torchpq/kernels/CustomKernel.py
torchpq/kernels/DistributedIVFPQTop1Cuda.py
torchpq/kernels/DistributedIVFPQTopkCuda.py
torchpq/kernels/GetAddressByIDCuda.py
torchpq/kernels/GetDivByAddressCuda.py
torchpq/kernels/GetDivByAddressV2Cuda.py
torchpq/kernels/GetIOACuda.py
torchpq/kernels/GetWriteAddressCuda.py
torchpq/kernels/GetWriteAddressV2Cuda.py
torchpq/kernels/IVFPQ4Top1Cuda_v1.py
torchpq/kernels/IVFPQTop1Cuda.py
torchpq/kernels/IVFPQTop1Cuda_v2.py
torchpq/kernels/IVFPQTopkCuda.py
torchpq/kernels/MBMMCuda.py
torchpq/kernels/MaxSimCuda.py
torchpq/kernels/MinBMMCuda.py
torchpq/kernels/PQDecodeCuda.py
torchpq/kernels/Top1SelectCuda.py
torchpq/kernels/Top32SelectCuda.py
torchpq/kernels/TopkBMMCuda.py
torchpq/kernels/TopkSelectCuda.py
torchpq/kernels/TopkSelectCuda_v2.py
torchpq/kernels/__init__.py
torchpq/kernels/cuda/bmm.cu
torchpq/kernels/cuda/bmm_helpers.cu
torchpq/kernels/cuda/compute_centroids.cu
torchpq/kernels/cuda/compute_product.cu
torchpq/kernels/cuda/distirbuted_ivfpq_top1.cu
torchpq/kernels/cuda/distributed_ivfpq_topk.cu
torchpq/kernels/cuda/get_address_by_id.cu
torchpq/kernels/cuda/get_div_by_address.cu
torchpq/kernels/cuda/get_div_by_address_v2.cu
torchpq/kernels/cuda/get_ioa.cu
torchpq/kernels/cuda/get_write_address.cu
torchpq/kernels/cuda/get_write_address_v2.cu
torchpq/kernels/cuda/ivfpq4_top1_v1.cu
torchpq/kernels/cuda/ivfpq4_top1_v1_failed.cu
torchpq/kernels/cuda/ivfpq_top1.cu
torchpq/kernels/cuda/ivfpq_top1_v2.cu
torchpq/kernels/cuda/ivfpq_top1_v2_failed.cu
torchpq/kernels/cuda/ivfpq_topk.cu
torchpq/kernels/cuda/ivfpq_topk_v1.cu
torchpq/kernels/cuda/ivfpq_topk_v2.cu
torchpq/kernels/cuda/max_sim.cu
torchpq/kernels/cuda/mbmm.cu
torchpq/kernels/cuda/minbmm.cu
torchpq/kernels/cuda/pq_decode.cu
torchpq/kernels/cuda/top1_select.cu
torchpq/kernels/cuda/top32_select.cu
torchpq/kernels/cuda/topk_select.cu
torchpq/kernels/cuda/topk_select_v2.cu
torchpq/kernels/cuda/topk_select_v2_maybe_later.cu
torchpq/kernels/cuda/topkbmm.cu
torchpq/legacy/IVFPQ.py
torchpq/legacy/IVFPQBase.py
torchpq/legacy/IVFPQR.py
torchpq/legacy/IVFPQTopk.py
torchpq/legacy/PQ.py
torchpq/legacy/SQ.py
torchpq/legacy/__init__.py
torchpq/transform/OPQ.py
torchpq/transform/PCA.py
torchpq/transform/__init__.py