Function hammingMatcherTest

test/hamming.cpp:46–95

template<typename T>
void hammingMatcherTest(string pTestFile, int feat_dim) {
    using af::dim4;

    vector<dim4> numDims;
    vector<vector<uint>> in32;
    vector<vector<uint>> tests;

    readTests<uint, uint, int>(pTestFile, numDims, in32, tests);

    vector<vector<T>> in(in32.size());
    for (size_t i = 0; i < in32[0].size(); i++) in[0].push_back((T)in32[0][i]);
    for (size_t i = 0; i < in32[1].size(); i++) in[1].push_back((T)in32[1][i]);

    dim4 qDims     = numDims[0];
    dim4 tDims     = numDims[1];
    af_array query = 0;
    af_array train = 0;
    af_array idx   = 0;
    af_array dist  = 0;

    ASSERT_SUCCESS(af_create_array(&query, &(in[0].front()), qDims.ndims(),
                                   qDims.get(),
                                   (af_dtype)dtype_traits<T>::af_type));
    ASSERT_SUCCESS(af_create_array(&train, &(in[1].front()), tDims.ndims(),
                                   tDims.get(),
                                   (af_dtype)dtype_traits<T>::af_type));

    ASSERT_SUCCESS(af_hamming_matcher(&idx, &dist, query, train, feat_dim, 1));

    vector<uint> goldIdx  = tests[0];
    vector<uint> goldDist = tests[1];
    size_t nElems         = goldIdx.size();
    uint *outIdx          = new uint[nElems];
    uint *outDist         = new uint[nElems];

    ASSERT_SUCCESS(af_get_data_ptr((void *)outIdx, idx));
    ASSERT_SUCCESS(af_get_data_ptr((void *)outDist, dist));

    for (size_t elIter = 0; elIter < nElems; ++elIter) {
        ASSERT_EQ(goldDist[elIter], outDist[elIter])
            << "at: " << elIter << endl;
    }

    delete[] outIdx;
    delete[] outDist;
    ASSERT_SUCCESS(af_release_array(query));
    ASSERT_SUCCESS(af_release_array(train));
    ASSERT_SUCCESS(af_release_array(idx));
    ASSERT_SUCCESS(af_release_array(dist));
}

TYPED_TEST(HammingMatcher8, Hamming_500_5000_Dim0) {
    UNSUPPORTED_BACKEND(AF_BACKEND_ONEAPI);