cub_gpu_tests.cu

/*
 * cub_gpu_tests.cu
 *
 *  Created on: May 15, 2019
 *      Author: i-bird
 */
 
#ifndef CUB_GPU_TESTS_CU_
#define CUB_GPU_TESTS_CU_
 
#include "config.h"
#define BOOST_TEST_DYN_LINK
#include <boost/test/unit_test.hpp>
#include "cub/cub.cuh"
#include "Vector/map_vector.hpp"
 
BOOST_AUTO_TEST_SUITE( cub_gpu_tests )
 
BOOST_AUTO_TEST_CASE( cub_gpu_scan_test )
{
    openfpm::vector_gpu<aggregate<unsigned int>> input;
    openfpm::vector_gpu<aggregate<unsigned int>> output;
 
    openfpm::vector_gpu<aggregate<unsigned char>> temporal;
 
    input.resize(10000);
    output.resize(10000);
 
    // fill input
 
    for (size_t i = 0 ; i < 10000; i++)
    {
        input.template get<0>(i) = 10.0*(float)rand() / RAND_MAX;
    }
 
    input.template hostToDevice<0>();
 
    void *d_temp_storage = NULL;
    size_t temp_storage_bytes = 0;
    cub::DeviceScan::ExclusiveSum(d_temp_storage, temp_storage_bytes,(unsigned int *)input.template getDeviceBuffer<0>(),
                                                                (unsigned int *)output.template getDeviceBuffer<0>(),
                                                                input.size());
 
    temporal.resize(temp_storage_bytes);
 
    // Run
    cub::DeviceScan::ExclusiveSum(temporal.template getDeviceBuffer<0>(), temp_storage_bytes,(unsigned int *)input.template getDeviceBuffer<0>(),
            (unsigned int *)output.template getDeviceBuffer<0>(),
            input.size());
 
    // Check
 
    output.template deviceToHost<0>();
 
    size_t cnt = 0;
    for (size_t i = 0 ; i < input.size() ; i++)
    {
        BOOST_REQUIRE_EQUAL(cnt,output.template get<0>(i));
        cnt += input.template get<0>(i);
    }
}
 
 
BOOST_AUTO_TEST_SUITE_END()
 
#endif /* CUB_GPU_TESTS_CU_ */