Commit ac76c308 authored by 蔡宇东's avatar 蔡宇东
Browse files

#89 update unittest 


Former-commit-id: 4edd9e6c6face5494ab4df20e3c4d799429d7b02
parent 2c47685b

core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp

+2 −2
Original line number Diff line number Diff line
@@ -541,15 +541,15 @@ void test_ivfsq8h_gpu(const std::string& ann_test_name, 
TEST(FAISSTEST, BENCHMARK) {

    test_ann_hdf5("sift-128-euclidean", "IVF4096,Flat",       2, {8, 128});

    test_ann_hdf5("sift-128-euclidean", "IVF16384,SQ8",       2, {8, 128});

    test_ann_hdf5("sift-128-euclidean", "IVF16384,SQ8Hybrid", 2, {8, 128});

#ifdef CUSTOMIZATION

    test_ann_hdf5("sift-128-euclidean", "IVF16384,SQ8Hybrid", 2, {8, 128});

    test_ivfsq8h_gpu("sift-128-euclidean", 2, {8, 128});

#endif



    test_ann_hdf5("glove-200-angular", "IVF4096,Flat",       1, {8, 128});

    test_ann_hdf5("glove-200-angular", "IVF16384,SQ8",       1, {8, 128});

    test_ann_hdf5("glove-200-angular", "IVF16384,SQ8Hybrid", 1, {8, 128});

#ifdef CUSTOMIZATION

    test_ann_hdf5("glove-200-angular", "IVF16384,SQ8Hybrid", 1, {8, 128});

    test_ivfsq8h_gpu("glove-200-angular", 2, {8, 128});

#endif

}