/* * Copyright 2008-2012 NVIDIA Corporation * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ /*! \file for_each.h * \brief Applies a function to each element in a range */ #pragma once #include #include #include namespace thrust { /*! \addtogroup modifying * \ingroup transformations * \{ */ /*! \p for_each applies the function object \p f to each element * in the range [first, last); \p f's return value, if any, * is ignored. Unlike the C++ Standard Template Library function * std::for_each, this version offers no guarantee on * order of execution. For this reason, this version of \p for_each * does not return a copy of the function object. * * The algorithm's execution is parallelized as determined by \p exec. * * \param exec The execution policy to use for parallelization. * \param first The beginning of the sequence. * \param last The end of the sequence. * \param f The function object to apply to the range [first, last). * \return last * * \tparam DerivedPolicy The name of the derived execution policy. * \tparam InputIterator is a model of Input Iterator, * and \p InputIterator's \c value_type is convertible to \p UnaryFunction's \c argument_type. * \tparam UnaryFunction is a model of Unary Function, * and \p UnaryFunction does not apply any non-constant operation through its argument. * * The following code snippet demonstrates how to use \p for_each to print the elements * of a \p std::device_vector using the \p thrust::device parallelization policy: * * \code * #include * #include * #include * #include * ... * * struct printf_functor * { * __host__ __device__ * void operator()(int x) * { * // note that using printf in a __device__ function requires * // code compiled for a GPU with compute capability 2.0 or * // higher (nvcc --arch=sm_20) * printf("%d\n"); * } * }; * ... * thrust::device_vector d_vec(3); * d_vec[0] = 0; d_vec[1] = 1; d_vec[2] = 2; * * thrust::for_each(thrust::device, d_vec.begin(), d_vec.end(), printf_functor()); * * // 0 1 2 is printed to standard output in some unspecified order * \endcode * * \see for_each_n * \see http://www.sgi.com/tech/stl/for_each.html */ template InputIterator for_each(const thrust::detail::execution_policy_base &exec, InputIterator first, InputIterator last, UnaryFunction f); /*! \p for_each_n applies the function object \p f to each element * in the range [first, first + n); \p f's return value, if any, * is ignored. Unlike the C++ Standard Template Library function * std::for_each, this version offers no guarantee on * order of execution. * * The algorithm's execution is parallelized as determined by \p exec. * * \param exec The execution policy to use for parallelization. * \param first The beginning of the sequence. * \param n The size of the input sequence. * \param f The function object to apply to the range [first, first + n). * \return first + n * * \tparam DerivedPolicy The name of the derived execution policy. * \tparam InputIterator is a model of Input Iterator, * and \p InputIterator's \c value_type is convertible to \p UnaryFunction's \c argument_type. * \tparam Size is an integral type. * \tparam UnaryFunction is a model of Unary Function, * and \p UnaryFunction does not apply any non-constant operation through its argument. * * The following code snippet demonstrates how to use \p for_each_n to print the elements * of a \p device_vector using the \p thrust::device parallelization policy. * * \code * #include * #include * #include * #include * * struct printf_functor * { * __host__ __device__ * void operator()(int x) * { * // note that using printf in a __device__ function requires * // code compiled for a GPU with compute capability 2.0 or * // higher (nvcc --arch=sm_20) * printf("%d\n"); * } * }; * ... * thrust::device_vector d_vec(3); * d_vec[0] = 0; d_vec[1] = 1; d_vec[2] = 2; * * thrust::for_each_n(thrust::device, d_vec.begin(), d_vec.size(), printf_functor()); * * // 0 1 2 is printed to standard output in some unspecified order * \endcode * * \see for_each * \see http://www.sgi.com/tech/stl/for_each.html */ template InputIterator for_each_n(const thrust::detail::execution_policy_base &exec, InputIterator first, Size n, UnaryFunction f); /*! \p for_each applies the function object \p f to each element * in the range [first, last); \p f's return value, if any, * is ignored. Unlike the C++ Standard Template Library function * std::for_each, this version offers no guarantee on * order of execution. For this reason, this version of \p for_each * does not return a copy of the function object. * * \param first The beginning of the sequence. * \param last The end of the sequence. * \param f The function object to apply to the range [first, last). * \return last * * \tparam InputIterator is a model of Input Iterator, * and \p InputIterator's \c value_type is convertible to \p UnaryFunction's \c argument_type. * \tparam UnaryFunction is a model of Unary Function, * and \p UnaryFunction does not apply any non-constant operation through its argument. * * The following code snippet demonstrates how to use \p for_each to print the elements * of a \p device_vector. * * \code * #include * #include * #include * * struct printf_functor * { * __host__ __device__ * void operator()(int x) * { * // note that using printf in a __device__ function requires * // code compiled for a GPU with compute capability 2.0 or * // higher (nvcc --arch=sm_20) * printf("%d\n"); * } * }; * ... * thrust::device_vector d_vec(3); * d_vec[0] = 0; d_vec[1] = 1; d_vec[2] = 2; * * thrust::for_each(d_vec.begin(), d_vec.end(), printf_functor()); * * // 0 1 2 is printed to standard output in some unspecified order * \endcode * * \see for_each_n * \see http://www.sgi.com/tech/stl/for_each.html */ template InputIterator for_each(InputIterator first, InputIterator last, UnaryFunction f); /*! \p for_each_n applies the function object \p f to each element * in the range [first, first + n); \p f's return value, if any, * is ignored. Unlike the C++ Standard Template Library function * std::for_each, this version offers no guarantee on * order of execution. * * \param first The beginning of the sequence. * \param n The size of the input sequence. * \param f The function object to apply to the range [first, first + n). * \return first + n * * \tparam InputIterator is a model of Input Iterator, * and \p InputIterator's \c value_type is convertible to \p UnaryFunction's \c argument_type. * \tparam Size is an integral type. * \tparam UnaryFunction is a model of Unary Function, * and \p UnaryFunction does not apply any non-constant operation through its argument. * * The following code snippet demonstrates how to use \p for_each_n to print the elements * of a \p device_vector. * * \code * #include * #include * #include * * struct printf_functor * { * __host__ __device__ * void operator()(int x) * { * // note that using printf in a __device__ function requires * // code compiled for a GPU with compute capability 2.0 or * // higher (nvcc --arch=sm_20) * printf("%d\n"); * } * }; * ... * thrust::device_vector d_vec(3); * d_vec[0] = 0; d_vec[1] = 1; d_vec[2] = 2; * * thrust::for_each_n(d_vec.begin(), d_vec.size(), printf_functor()); * * // 0 1 2 is printed to standard output in some unspecified order * \endcode * * \see for_each * \see http://www.sgi.com/tech/stl/for_each.html */ template InputIterator for_each_n(InputIterator first, Size n, UnaryFunction f); /*! \} // end modifying */ } // end namespace thrust #include