Function TEST

test/nearest_neighbour.cpp:185–220

//////////////////////////////// CPP ////////////////////////////////

///////////////////////////////////// CPP ////////////////////////////////
//
TEST(NearestNeighbourSSD, CPP) {
    UNSUPPORTED_BACKEND(AF_BACKEND_ONEAPI);
    vector<dim4> numDims;
    vector<vector<uint>> in;
    vector<vector<uint>> tests;

    readTests<uint, uint, uint>(TEST_DIR
                                "/nearest_neighbour/ssd_500_5000_dim0.test",
                                numDims, in, tests);

    dim4 qDims = numDims[0];
    dim4 tDims = numDims[1];

    array query(qDims, &(in[0].front()));
    array train(tDims, &(in[1].front()));

    array idx, dist;
    nearestNeighbour(idx, dist, query, train, 0, 1, AF_SSD);

    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];

    idx.host(outIdx);
    dist.host(outDist);

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

    delete[] outIdx;
    delete[] outDist;
}

TEST(NearestNeighbourSAD, CPP) {
    UNSUPPORTED_BACKEND(AF_BACKEND_ONEAPI);