~ubuntu-branches/ubuntu/trusty/libthrust/trusty

« back to all changes in this revision

Viewing changes to detail/device/dispatch/sort.h

  • Committer: Bazaar Package Importer
  • Author(s): Andreas Beckmann
  • Date: 2011-05-28 09:32:48 UTC
  • Revision ID: james.westby@ubuntu.com-20110528093248-np3euv5sj7fw3nyv
Tags: upstream-1.4.0
ImportĀ upstreamĀ versionĀ 1.4.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/*
 
2
 *  Copyright 2008-2011 NVIDIA Corporation
 
3
 *
 
4
 *  Licensed under the Apache License, Version 2.0 (the "License");
 
5
 *  you may not use this file except in compliance with the License.
 
6
 *  You may obtain a copy of the License at
 
7
 *
 
8
 *      http://www.apache.org/licenses/LICENSE-2.0
 
9
 *
 
10
 *  Unless required by applicable law or agreed to in writing, software
 
11
 *  distributed under the License is distributed on an "AS IS" BASIS,
 
12
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
13
 *  See the License for the specific language governing permissions and
 
14
 *  limitations under the License.
 
15
 */
 
16
 
 
17
#pragma once
 
18
 
 
19
#include <thrust/iterator/iterator_traits.h>
 
20
 
 
21
#include <thrust/detail/device/cuda/sort.h>
 
22
#include <thrust/detail/device/omp/sort.h>
 
23
 
 
24
namespace thrust
 
25
{
 
26
namespace detail
 
27
{
 
28
namespace device
 
29
{
 
30
namespace dispatch
 
31
{
 
32
 
 
33
template<typename RandomAccessIterator,
 
34
         typename StrictWeakOrdering>
 
35
  void stable_sort(RandomAccessIterator first,
 
36
                   RandomAccessIterator last,
 
37
                   StrictWeakOrdering comp,
 
38
                   thrust::detail::omp_device_space_tag)
 
39
{
 
40
    // OpenMP implementation
 
41
    thrust::detail::device::omp::stable_sort(first, last, comp);
 
42
}
 
43
 
 
44
template<typename RandomAccessIterator,
 
45
         typename StrictWeakOrdering>
 
46
  void stable_sort(RandomAccessIterator first,
 
47
                   RandomAccessIterator last,
 
48
                   StrictWeakOrdering comp,
 
49
                   thrust::detail::cuda_device_space_tag)
 
50
{
 
51
    // CUDA implementation
 
52
    thrust::detail::device::cuda::stable_sort(first, last, comp);
 
53
}
 
54
 
 
55
template<typename RandomAccessKeyIterator,
 
56
         typename RandomAccessValueIterator,
 
57
         typename StrictWeakOrdering>
 
58
  void stable_sort_by_key(RandomAccessKeyIterator keys_first,
 
59
                          RandomAccessKeyIterator keys_last,
 
60
                          RandomAccessValueIterator values_first,
 
61
                          StrictWeakOrdering comp,
 
62
                          thrust::detail::omp_device_space_tag,
 
63
                          thrust::detail::omp_device_space_tag)
 
64
{
 
65
    // OpenMP implementation
 
66
    thrust::detail::device::omp::stable_sort_by_key(keys_first, keys_last, values_first, comp);
 
67
}
 
68
 
 
69
template<typename RandomAccessKeyIterator,
 
70
         typename RandomAccessValueIterator,
 
71
         typename StrictWeakOrdering>
 
72
  void stable_sort_by_key(RandomAccessKeyIterator keys_first,
 
73
                          RandomAccessKeyIterator keys_last,
 
74
                          RandomAccessValueIterator values_first,
 
75
                          StrictWeakOrdering comp,
 
76
                          thrust::detail::cuda_device_space_tag,
 
77
                          thrust::detail::cuda_device_space_tag)
 
78
{
 
79
    // CUDA implementation
 
80
    thrust::detail::device::cuda::stable_sort_by_key(keys_first, keys_last, values_first, comp);
 
81
}
 
82
 
 
83
} // end namespace dispatch
 
84
} // end namespace device
 
85
} // end namespace detail
 
86
} // end namespace thrust
 
87