1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/*
* Copyright 2008-2013 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.
*/
#pragma once
#include <thrust/detail/config.h>
#include <thrust/system/omp/detail/reduce_intervals.h>
#include <thrust/iterator/iterator_traits.h>
#include <thrust/detail/function.h>
#include <thrust/detail/cstdint.h>
THRUST_NAMESPACE_BEGIN
namespace system
{
namespace omp
{
namespace detail
{
template <typename DerivedPolicy,
typename InputIterator,
typename OutputIterator,
typename BinaryFunction,
typename Decomposition>
void reduce_intervals(execution_policy<DerivedPolicy> &,
InputIterator input,
OutputIterator output,
BinaryFunction binary_op,
Decomposition decomp)
{
// we're attempting to launch an omp kernel, assert we're compiling with omp support
// ========================================================================
// X Note to the user: If you've found this line due to a compiler error, X
// X you need to enable OpenMP support in your compiler. X
// ========================================================================
THRUST_STATIC_ASSERT_MSG(
(thrust::detail::depend_on_instantiation<
InputIterator, (THRUST_DEVICE_COMPILER_IS_OMP_CAPABLE == THRUST_TRUE)
>::value)
, "OpenMP compiler support is not enabled"
);
#if (THRUST_DEVICE_COMPILER_IS_OMP_CAPABLE == THRUST_TRUE)
typedef typename thrust::iterator_value<OutputIterator>::type OutputType;
// wrap binary_op
thrust::detail::wrapped_function<BinaryFunction,OutputType> wrapped_binary_op(binary_op);
typedef thrust::detail::intptr_t index_type;
index_type n = static_cast<index_type>(decomp.size());
THRUST_PRAGMA_OMP(parallel for)
for(index_type i = 0; i < n; i++)
{
InputIterator begin = input + decomp[i].begin();
InputIterator end = input + decomp[i].end();
if (begin != end)
{
OutputType sum = thrust::raw_reference_cast(*begin);
++begin;
while (begin != end)
{
sum = wrapped_binary_op(sum, *begin);
++begin;
}
OutputIterator tmp = output + i;
*tmp = sum;
}
}
#endif // THRUST_DEVICE_COMPILER_IS_OMP_CAPABLE
}
} // end namespace detail
} // end namespace omp
} // end namespace system
THRUST_NAMESPACE_END
|
