ceph/src/boost/libs/compute/perf/perf_rotate_copy.cpp

   1 //---------------------------------------------------------------------------//
   2 // Copyright (c) 2014 Roshan <thisisroshansmail@gmail.com>
   3 //
   4 // Distributed under the Boost Software License, Version 1.0
   5 // See accompanying file LICENSE_1_0.txt or copy at
   6 // http://www.boost.org/LICENSE_1_0.txt
   7 //
   8 // See http://boostorg.github.com/compute for more information.
   9 //---------------------------------------------------------------------------//
  10
  11 #include <algorithm>
  12 #include <iostream>
  13 #include <numeric>
  14 #include <vector>
  15
  16 #include <boost/compute/system.hpp>
  17 #include <boost/compute/algorithm/rotate_copy.hpp>
  18 #include <boost/compute/container/vector.hpp>
  19
  20 #include "perf.hpp"
  21
  22 int rand_int()
  23 {
  24     return static_cast<int>((rand() / double(RAND_MAX)) * 25.0);
  25 }
  26
  27 int main(int argc, char *argv[])
  28 {
  29     perf_parse_args(argc, argv);
  30     std::cout << "size: " << PERF_N << std::endl;
  31
  32     // setup context and queue for the default device
  33     boost::compute::device device = boost::compute::system::default_device();
  34     boost::compute::context context(device);
  35     boost::compute::command_queue queue(context, device);
  36     std::cout << "device: " << device.name() << std::endl;
  37
  38     // create vector of random numbers on the host
  39     std::vector<int> host_vector(PERF_N);
  40     std::generate(host_vector.begin(), host_vector.end(), rand_int);
  41
  42     // create vector on the device and copy the data
  43     boost::compute::vector<int> device_vector(PERF_N, context);
  44     boost::compute::copy(
  45         host_vector.begin(), host_vector.end(), device_vector.begin(), queue
  46     );
  47
  48     boost::compute::vector<int> device_vector2(PERF_N, context);
  49
  50     perf_timer t;
  51     for(size_t trial = 0; trial < PERF_TRIALS; trial++){
  52         t.start();
  53         boost::compute::rotate_copy(
  54             device_vector.begin(), device_vector.begin()+(PERF_N/2), device_vector.end(), device_vector2.begin(), queue
  55         );
  56         queue.finish();
  57         t.stop();
  58     }
  59     std::cout << "time: " << t.min_time() / 1e6 << " ms" << std::endl;
  60
  61     return 0;
  62 }