提供一个完整代码说明如何使用omp_get_thread_num()
时间: 2023-12-02 21:05:16 浏览: 119
`omp_get_thread_num()`是OpenMP库中的一个函数,用于获取当前线程的线程号。下面是一个使用`omp_get_thread_num()`函数的示例代码,该代码使用OpenMP库实现了并行计算数组元素总和的功能。
```c
#include <stdio.h>
#include <omp.h>
int main() {
int sum = 0;
int n = 100;
int arr[n];
// 初始化数组
for(int i = 0; i < n; i++) {
arr[i] = i;
}
#pragma omp parallel num_threads(4)
{
int local_sum = 0;
int tid = omp_get_thread_num();
// 计算每个线程负责的部分
int start = tid * n / 4;
int end = (tid + 1) * n / 4;
for(int i = start; i < end; i++) {
local_sum += arr[i];
}
// 同步线程的局部和
#pragma omp critical
sum += local_sum;
}
printf("总和为:%d\n", sum);
return 0;
}
```
在上面的代码中,使用`#pragma omp parallel`指令创建了一个并行区域,其中`num_threads(4)`指令设置了使用4个线程。在并行区域内,每个线程使用`omp_get_thread_num()`函数获取自己的线程号,然后计算出自己负责的部分,并将结果保存在`local_sum`中。最后,使用`#pragma omp critical`指令同步各个线程的局部和,得到最终的结果。
阅读全文