码迷,mamicode.com
首页 > 编程语言 > 详细

通过打包 accumulate 实现多线程版本的 accumulate

时间:2015-02-18 00:55:12      阅读:172      评论:0      收藏:0      [点我收藏+]

标签:

#include <iostream>
#include <algorithm>
#include <thread>
#include <functional>

using namespace std;

template <typename Iterator, typename T>
struct accumulate_block
{
  void operator ()(Iterator first, Iterator last, T &result)
  {
    result = accumulate(first, last, result);
  }
};

template <typename Iterator, typename T>
T parallel_accumulate(Iterator first, Iterator last, T init)
{
    unsigned long const length = std::distance(first, last);

//if there is no element, return init value
    if(!length){
        return init;
    }


    unsigned long const min_per_thread = 25;

    //how much threads at least we needed
    unsigned long const max_threads =
        (length + min_per_thread - 1) / min_per_thread;

    unsigned long const hardware_threads =
        std::thread::hardware_concurrency();

//if max_threads more than hardware threads,
//just use hardware threads
//if hardware threads not support even muiltithreads, use 2 threads
//instead of , use hardware threads
    unsigned long const num_threads =
        std::min( hardware_threads != 0
                                                         ? hardware_threads
                                                         :  2
                     , max_threads);

    //the length of the range was divided by threads
    unsigned long const block_size = length / num_threads;

    std::vector<T>                 results  (num_threads);
    std::vector<std::thread>  threads (num_threads - 1);

    auto block_start = first;
    for(unsigned long i = 0; i < (num_threads - 1); ++i){
        auto block_end = block_start;
        
        //put block_end to the end of current block
        std::advance(block_end, block_size);
        
        threads[i] = std::thread(
              accumulate_block<Iterator, T>(),
              block_start, block_end, std::ref(results[i]));
        block_start = block_end;
    }
        accumulate_block<Iterator, T>()(block_start,
                                         last,
                                         results[num_threads - 1]);
        std::for_each(threads.begin(), threads.end(),
                                 std::mem_fn(&std::thread::join));

        return std::accumulate(results.begin(), results.end(), init);
}

int main()
{
    vector<int> ivec{1, 2, 3, 4, 5, 5, 6, 7, 8, 9};
    int result = 0;
    result = parallel_accumulate(ivec.cbegin(), ivec.cend(), result);
    cout << result << endl;
    return 0;
}

 

通过打包 accumulate 实现多线程版本的 accumulate

标签:

原文地址:http://www.cnblogs.com/wuOverflow/p/4295566.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!