当在另一个 GPU 而不是 0 中运行时，带有 cuda 的 openMP 会出现分段错误

发布于 2024-12-12 10:07:48 字数 394 浏览 0 评论 0原文

我正在使用 cuda 并使用 openMP 进行并行线程：

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

    if(cpu_thread_id==0) 
    {
        // call kernel function
    }
}

即使我有 4 个 GPU，它也始终使用 cpu_thread_id=0 工作。所以当我将cpu_thread_id更改为另一个数字1,2和3进行测试时。我得到：分段错误。出现这个错误，我不知道为什么会出现这个问题。

你注意到什么了吗？

提前致谢

原文

I am working with cuda and using openMP for parallel threading:

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

    if(cpu_thread_id==0) 
    {
        // call kernel function
    }
}

It is always working with cpu_thread_id=0 even I have 4 gpus. So when I change the cpu_thread_id to another number 1,2, and 3 for testing. I got: segmentation fault. with this error, I have no idea why it the problem.

Did you notice any thing?

Thanks in advance

分享到QQ

分享到微博