mirrors
/
faiss
mirror of https://github.com/facebookresearch/faiss.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add test to cover GPU (#4130) 

Summary:

same as title

Differential Revision: D68388863
export-D68388863
Junjie Qi 2025-01-19 00:49:41 -08:00 committed by Facebook GitHub Bot
parent 3ed52b001c
commit 7e7f94d738
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_contrib.py
View File

@ -35,7 +35,7 @@ from faiss.contrib.ondisk import merge_ondisk
class TestComputeGT(unittest.TestCase): class TestComputeGT(unittest.TestCase):
def do_test_compute_GT(self, metric=faiss.METRIC_L2): def do_test_compute_GT(self, metric=faiss.METRIC_L2, ngpu=0):
d = 64 d = 64
xt, xb, xq = get_dataset_2(d, 0, 10000, 100) xt, xb, xq = get_dataset_2(d, 0, 10000, 100)
@ -50,7 +50,7 @@ class TestComputeGT(unittest.TestCase):
yield xb[i0:i0 + bs] yield xb[i0:i0 + bs]
Dnew, Inew = knn_ground_truth( Dnew, Inew = knn_ground_truth(
xq, matrix_iterator(xb, 1000), 10, metric, ngpu=0) xq, matrix_iterator(xb, 1000), 10, metric, ngpu=ngpu)
np.testing.assert_array_equal(Iref, Inew) np.testing.assert_array_equal(Iref, Inew)
# decimal = 4 required when run on GPU # decimal = 4 required when run on GPU
@ -62,6 +62,12 @@ class TestComputeGT(unittest.TestCase):
def test_compute_GT_ip(self): def test_compute_GT_ip(self):
self.do_test_compute_GT(faiss.METRIC_INNER_PRODUCT) self.do_test_compute_GT(faiss.METRIC_INNER_PRODUCT)
def test_compute_GT_gpu(self):
self.do_test_compute_GT(ngpu=-1)
def test_compute_GT_ip_gpu(self):
self.do_test_compute_GT(faiss.METRIC_INNER_PRODUCT, ngpu=-1)
class TestDatasets(unittest.TestCase): class TestDatasets(unittest.TestCase):
"""here we test only the synthetic dataset. Datasets that require """here we test only the synthetic dataset. Datasets that require