openMP 只运行一个线程?

发布于 2024-12-09 08:23:57 字数 1104 浏览 0 评论 0原文

我在 main:

int main(int argv, char **argc)
{
   // Get multi-CPU/multi-GPU data
   int num_gpus;
   cudaGetDeviceCount(&num_gpus);
   printf("### Number of host CPUs:\t%d\n", omp_get_num_procs());
   printf("### Number of CUDA devices:\t%d\n", num_gpus);
   omp_set_num_threads(num_gpus);

   #pragma omp parallel
   {    
    unsigned int cpu_thread_id = omp_get_thread_num();  

    printf("### (CPU thread %d)\n",cpu_thread_id);  

    test(cpu_thread_id, num_gpus);
   }    
}

和测试函数中有这段代码:

void test(unsigned int cpu_thread_id, int num_gpus)
{   
    printf("### Using CUDA device %d, GPU = %d\n", cpu_thread_id, num_gpus);
}

即使我得到了输出:

### Number of host CPUs:        8
### Number of CUDA devices:     4
### (CPU thread 0)
### Using CUDA device 0, GPU = 4

但我期望有更多像这样的线程:

### (CPU thread 1)
### Using CUDA device 1, GPU = 4
### (CPU thread 2)
### Using CUDA device 2, GPU = 4
### (CPU thread 3)
### Using CUDA device 3, GPU = 4

为什么其他三个线程没有运行?

提前致谢。

I have this code in the main:

int main(int argv, char **argc)
{
   // Get multi-CPU/multi-GPU data
   int num_gpus;
   cudaGetDeviceCount(&num_gpus);
   printf("### Number of host CPUs:\t%d\n", omp_get_num_procs());
   printf("### Number of CUDA devices:\t%d\n", num_gpus);
   omp_set_num_threads(num_gpus);

   #pragma omp parallel
   {    
    unsigned int cpu_thread_id = omp_get_thread_num();  

    printf("### (CPU thread %d)\n",cpu_thread_id);  

    test(cpu_thread_id, num_gpus);
   }    
}

and the test function:

void test(unsigned int cpu_thread_id, int num_gpus)
{   
    printf("### Using CUDA device %d, GPU = %d\n", cpu_thread_id, num_gpus);
}

Even I got the output:

### Number of host CPUs:        8
### Number of CUDA devices:     4
### (CPU thread 0)
### Using CUDA device 0, GPU = 4

But I expected to have more threads like this:

### (CPU thread 1)
### Using CUDA device 1, GPU = 4
### (CPU thread 2)
### Using CUDA device 2, GPU = 4
### (CPU thread 3)
### Using CUDA device 3, GPU = 4

Why the other three threads are not running?

Thanks in advance.

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文