CUDA 计时

最新推荐文章于 2025-01-03 09:43:11 发布

转载最新推荐文章于 2025-01-03 09:43:11 发布 · 83 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/superlc320/p/3932609.html

文章标签：

#python

CUDA Runtime API提供了用于计时的接口，可以用如下代码对在GPU上运行的CUDA程序进行计时，计时结果以毫秒为单位：

cudaEvent_t start, stop;  
cudaEventCreate(&start);      
cudaEventCreate(&stop);    
cudaEventRecord(start, 0);  
  
//需要计时的在GPU上运行的程序  

cudaEventRecord(stop, 0);   
cudaEventSynchronize(stop);  
float elapsedTime;  
cudaEventElapsedTime(&elapsedTime, start, stop);  
cudaEventDestroy(start);  
cudaEventDestroy(stop);  
printf("Kernel time: %.3f ms\n", elapsedTime);

需要注意的是函数cudaEventSynchronize() 不可或缺，因为CUDA的kernel函数是以异步方式执行的，调用后立刻返回，这会导致计时不准确。

cudaEventSynchronize(stop)会使得直到GPU执行完cudaEventRecord(stop, 0)之前的所有语句时，事件stop才会被记录下来，即起到同步的作用。

转载于:https://www.cnblogs.com/superlc320/p/3932609.html