基于Jacobi迭代的OpenMP 和MPI优化并行

46 篇文章 29 订阅 ¥239.90 ¥99.00

问题描述

求解微分方程:
φ x x + φ y y − α φ = F . \varphi_{xx} + \varphi_{yy} - \alpha \varphi = F. φ

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
以下是使用Jacobi代,多线程并行计算二维卷积的C++代码,不使用OpenMPMPI: ```c++ #include <iostream> #include <thread> #include <vector> using namespace std; const int N = 1000; // 矩阵大小 const int K = 3; // 卷积核大小 const int P = (K - 1) / 2; // 边界填充大小 const int num_threads = 4; // 线程数 // 边界填充 void padding(float in[N][N], float pad[N + 2 * P][N + 2 * P]) { for (int i = 0; i < N + 2 * P; ++i) { for (int j = 0; j < N + 2 * P; ++j) { if (i < P || i >= N + P || j < P || j >= N + P) { pad[i][j] = 0; } else { pad[i][j] = in[i - P][j - P]; } } } } // 卷积核计算 void conv(float pad[N + 2 * P][N + 2 * P], float out[N][N], int i, int j) { float sum = 0; for (int u = -P; u <= P; ++u) { for (int v = -P; v <= P; ++v) { sum += pad[i + P + u][j + P + v]; } } out[i][j] = sum / (K * K); } // Jacobi代 void jacobi(float in[N][N], float out[N][N], int start, int end) { float pad[N + 2 * P][N + 2 * P]; padding(in, pad); for (int i = start; i < end; ++i) { for (int j = 0; j < N; ++j) { conv(pad, out, i, j); } } } int main() { float in[N][N], out[N][N]; // 初始化输入矩阵 for (int i = 0; i < N; ++i) { for (int j = 0; j < N; ++j) { in[i][j] = i * j; } } vector<thread> threads; int block_size = N / num_threads; // 创建线程 for (int t = 0; t < num_threads; ++t) { int start = t * block_size; int end = (t + 1) * block_size; if (t == num_threads - 1) { end = N; } threads.emplace_back(jacobi, in, out, start, end); } // 等待线程结束 for (auto &t : threads) { t.join(); } // 输出结果矩阵 for (int i = 0; i < N; ++i) { for (int j = 0; j < N; ++j) { cout << out[i][j] << " "; } cout << endl; } return 0; } ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Galerkin码农选手

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值