123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- //---------------------------------------------------------------------------//
- // Copyright (c) 2013-2014 Kyle Lutz <kyle.r.lutz@gmail.com>
- //
- // Distributed under the Boost Software License, Version 1.0
- // See accompanying file LICENSE_1_0.txt or copy at
- // http://www.boost.org/LICENSE_1_0.txt
- //
- // See http://boostorg.github.com/compute for more information.
- //---------------------------------------------------------------------------//
- #include <algorithm>
- #include <iostream>
- #include <vector>
- #include <boost/compute/system.hpp>
- #include <boost/compute/algorithm/copy.hpp>
- #include <boost/compute/algorithm/transform.hpp>
- #include <boost/compute/container/vector.hpp>
- namespace compute = boost::compute;
- // this example shows how to embed PTX assembly instructions
- // directly into boost.compute functions and use them with the
- // transform() algorithm.
- int main()
- {
- // get default device and setup context
- compute::device gpu = compute::system::default_device();
- compute::context context(gpu);
- compute::command_queue queue(context, gpu);
- std::cout << "device: " << gpu.name() << std::endl;
- // check to ensure we have an nvidia device
- if(gpu.vendor() != "NVIDIA Corporation"){
- std::cerr << "error: inline PTX assembly is only supported "
- << "on NVIDIA devices."
- << std::endl;
- return 0;
- }
- // create input values and copy them to the device
- using compute::uint_;
- uint_ data[] = { 0x00, 0x01, 0x11, 0xFF };
- compute::vector<uint_> input(data, data + 4, queue);
- // function returning the number of bits set (aka population count or
- // popcount) using the "popc" inline ptx assembly instruction.
- BOOST_COMPUTE_FUNCTION(uint_, nvidia_popc, (uint_ x),
- {
- uint count;
- asm("popc.b32 %0, %1;" : "=r"(count) : "r"(x));
- return count;
- });
- // calculate the popcount for each input value
- compute::vector<uint_> output(input.size(), context);
- compute::transform(
- input.begin(), input.end(), output.begin(), nvidia_popc, queue
- );
- // copy results back to the host and print them out
- std::vector<uint_> counts(output.size());
- compute::copy(output.begin(), output.end(), counts.begin(), queue);
- for(size_t i = 0; i < counts.size(); i++){
- std::cout << "0x" << std::hex << data[i]
- << " has " << counts[i]
- << " bits set" << std::endl;
- }
- return 0;
- }
|