12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- //---------------------------------------------------------------------------//
- // Copyright (c) 2013 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.
- //---------------------------------------------------------------------------//
- //[transform_sqrt_example
- #include <vector>
- #include <algorithm>
- #include <boost/compute/algorithm/transform.hpp>
- #include <boost/compute/container/vector.hpp>
- #include <boost/compute/functional/math.hpp>
- namespace compute = boost::compute;
- int main()
- {
- // get default device and setup context
- compute::device device = compute::system::default_device();
- compute::context context(device);
- compute::command_queue queue(context, device);
- // generate random data on the host
- std::vector<float> host_vector(10000);
- std::generate(host_vector.begin(), host_vector.end(), rand);
- // create a vector on the device
- compute::vector<float> device_vector(host_vector.size(), context);
- // transfer data from the host to the device
- compute::copy(
- host_vector.begin(), host_vector.end(), device_vector.begin(), queue
- );
- // calculate the square-root of each element in-place
- compute::transform(
- device_vector.begin(),
- device_vector.end(),
- device_vector.begin(),
- compute::sqrt<float>(),
- queue
- );
- // copy values back to the host
- compute::copy(
- device_vector.begin(), device_vector.end(), host_vector.begin(), queue
- );
- return 0;
- }
- //]
|