cudaSetDevice和线程

最新推荐文章于 2025-04-27 21:25:00 发布

转载最新推荐文章于 2025-04-27 21:25:00 发布 · 2.9k 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/reedlau/p/4949804.html

文章标签：

#c# #c/c++

本文讨论了CUDA编程中CUDASetDevice函数的线程安全性，并通过代码示例展示了新创建线程默认使用device0的情况。强调了始终设置当前设备的重要性以避免多线程错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.cudaSetDevice是线程安全的

2.新创建的线程默认是device 0

#include <stdio.h>
#include <pthread.h>
#include <cuda.h>

#define N_THREAD 2

void *thread_run(void *pp)
{
    int *p=(int*)pp;
    
    int tid=p[0];
    int id;

    cudaGetDevice(&id);
    printf("%d idev=%d\n",tid,id);
    
    
    pthread_exit(NULL);
}

int main(int argc,char *argv[])
{
    pthread_t tid[2];
    int i;
    int ii[2];
    
    cudaSetDevice(1);
    
    for (i=0;i<N_THREAD;++i)
    {
        ii[i]=i;
        pthread_create(&tid[i],NULL,thread_run,&ii[i]);
    }
    
    for (i=0;i<N_THREAD;++i)
    {
        pthread_join(tid[i],NULL);
    }
    
}