并行版的std::accumulate

2023-12-21 00:53:22

代码来自C++并发编程实战?

#include <iostream>
#include <thread>
#include<functional>
#include <algorithm>
#include <vector>
#include <numeric>//标准库里的accumulate在这个头文件里
using namespace std;
using namespace placeholders;
template<typename Iterator,typename T>
struct accumulage_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 = distance(first, last);//获得两个迭代器之间的距离
    if (!length)//长度为0,直接返回
        return init;
    unsigned long const min_per_thread = 25;
    unsigned long const max_threads = (length + min_per_thread - 1) / min_per_thread;
    unsigned long const hardware_threads = thread::hardware_concurrency();
    unsigned long const num_threads=min(hardware_threads!=0?hardware_threads:2, max_threads);//上面的是在决定线程数量
    unsigned long const block_size = length / num_threads;//根据线程数量把范围内的元素分块,每个线程处理对应块范围块的累加
    vector<T> results(num_threads);//每个线程对应的块范围的累加结果,存在这里
    vector<thread> threads(num_threads - 1);//线程数量-1,因为这里是主线程还算一个
    Iterator block_start = first;
    for (unsigned long i = 0; i < num_threads - 1; i++) {
        Iterator block_end = block_start;
        advance(block_end, block_size);//每次尾迭代器移动一个块的大小
        threads[i] = thread(accumulage_block<Iterator, T>(), block_start, block_end, ref(results[i]));//创建一个线程,计算一个块范围内的累加和
        block_start = block_end;//首迭代器移动,使得bolck_start和block_end始终相差一个块的距离
    }
    accumulage_block<Iterator, T>()(block_start, last, results[num_threads - 1]);//主线程处理 最后一块
    for_each(threads.begin(), threads.end(), mem_fn(&thread::join));//等待所有线程结束
    return accumulate(results.begin(), results.end(), init);//把每个线程的结果累加
}
int main()
{
    vector<int> vec{ 1,4,5,6,3,2 };
    int ans = parallel_accumulate<vector<int>::iterator,int>(vec.begin(),vec.end(), 0);
    cout << ans;
    
}

std::thread::hardware_concurrency 是 C++ 标准库中的一个函数,用于获取计算机硬件支持的并发线程数目。这个函数返回一个 unsigned int 类型的值,表示硬件支持的并发线程数,如果无法检测则返回 0。

文章来源:https://blog.csdn.net/ouliten/article/details/135118738
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。