| 22 | #include "perf.hpp" |
| 23 | |
| 24 | int main(int argc, char *argv[]) |
| 25 | { |
| 26 | perf_parse_args(argc, argv); |
| 27 | std::cout << "size: " << PERF_N << std::endl; |
| 28 | |
| 29 | // setup context and queue for the default device |
| 30 | boost::compute::device device = boost::compute::system::default_device(); |
| 31 | boost::compute::context context(device); |
| 32 | boost::compute::command_queue queue(context, device); |
| 33 | std::cout << "device: " << device.name() << std::endl; |
| 34 | |
| 35 | // create vector of random numbers on the host |
| 36 | std::vector<int> random_vector(PERF_N); |
| 37 | std::generate(random_vector.begin(), random_vector.end(), rand); |
| 38 | |
| 39 | // create input vector for gpu |
| 40 | std::vector<int> gpu_vector = random_vector; |
| 41 | |
| 42 | // sort vector on gpu |
| 43 | boost::timer::cpu_timer t; |
| 44 | boost::compute::sort( |
| 45 | gpu_vector.begin(), gpu_vector.end(), queue |
| 46 | ); |
| 47 | queue.finish(); |
| 48 | std::cout << "time: " << t.elapsed().wall / 1e6 << " ms" << std::endl; |
| 49 | |
| 50 | // create input vector for host |
| 51 | std::vector<int> host_vector = random_vector; |
| 52 | |
| 53 | // sort vector on host |
| 54 | t.start(); |
| 55 | std::sort(host_vector.begin(), host_vector.end()); |
| 56 | std::cout << "host time: " << t.elapsed().wall / 1e6 << " ms" << std::endl; |
| 57 | |
| 58 | // ensure that both sorted vectors are equal |
| 59 | if(!std::equal(gpu_vector.begin(), gpu_vector.end(), host_vector.begin())){ |
| 60 | std::cerr << "ERROR: sorted vectors not the same" << std::endl; |
| 61 | return -1; |
| 62 | } |
| 63 | |
| 64 | return 0; |
| 65 | } |