C++以其高效执行和低延迟特性,在高性能计算和科学计算领域被广泛应用。本文结合代码示例,讲解C++多线程和异步任务在高性能计算中的实战方法。

一、基础多线程计算

使用std::thread创建多线程任务:

#include <iostream>
#include <thread>
#include <vector>

void compute(int id) {
    std::cout << "任务 " << id << " 开始" << std::endl;
    int sum = 0;
    for(int i = 0; i < 1000000; i++) sum += i;
    std::cout << "任务 " << id << " 完成, 结果: " << sum << std::endl;
}

int main() {
    std::vector<std::thread> threads;
    for(int i = 1; i <= 5; i++) {
        threads.push_back(std::thread(compute, i));
    }
    for(auto &t : threads) t.join();
}

二、异步任务实现

利用std::async进行异步计算,提高CPU利用率:

#include <future>
#include <iostream>

int heavyCompute(int id) {
    int sum = 0;
    for(int i = 0; i < 1000000; i++) sum += i;
    return sum;
}

int main() {
    std::vector<std::future<int>> futures;
    for(int i = 1; i <= 5; i++) {
        futures.push_back(std::async(std::launch::async, heavyCompute, i));
    }
    for(int i = 0; i < futures.size(); i++) {
        std::cout << "任务 " << i+1 << " 结果: " << futures[i].get() << std::endl;
    }
}

三、线程池优化

自定义线程池管理多任务执行,提升资源利用率:

#include <queue>
#include <thread>
#include <mutex>
#include <condition_variable>
#include <functional>

class ThreadPool {
    std::vector<std::thread> workers;
    std::queue<std::function<void()>> tasks;
    std::mutex queue_mutex;
    std::condition_variable condition;
    bool stop = false;
public:
    ThreadPool(size_t threads) {
        for(size_t i = 0; i < threads; ++i)
            workers.emplace_back([this]{
                while(true) {
                    std::function<void()> task;
                    {
                        std::unique_lock<std::mutex> lock(this->queue_mutex);
                        this->condition.wait(lock, [this]{ return this->stop || !this->tasks.empty(); });
                        if(this->stop && this->tasks.empty()) return;
                        task = std::move(this->tasks.front());
                        this->tasks.pop();
                    }
                    task();
                }
            });
    }
    template<class F>
    void enqueue(F&& f) {
        {
            std::unique_lock<std::mutex> lock(queue_mutex);
            tasks.emplace(std::forward<F>(f));
        }
        condition.notify_one();
    }
    ~ThreadPool() {
        {
            std::unique_lock<std::mutex> lock(queue_mutex);
            stop = true;
        }
        condition.notify_all();
        for(std::thread &worker : workers) worker.join();
    }
};

四、高性能优化技巧

  1. 合理线程数量:根据CPU核心数调优线程池大小。

  2. 任务拆分:将大任务拆分为小块,充分利用并行计算能力。

  3. 异步计算结合缓存:避免重复计算,提高整体吞吐量。

  4. 性能监控:记录任务执行时间和资源使用情况,优化调度策略。

五、总结

C++结合多线程、异步任务和线程池,可以实现高性能计算系统。通过合理拆分任务、异步计算和线程池管理,开发者能够构建高效、稳定的计算框架,广泛应用于科学计算、金融分析

Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐