Kernel函数执行时间统计

Kernel函数是执行在GPU上的函数,并且是异步执行,所以统计其执行时间与CPU函数有不同之处。通常有三种方式可以用于kernel函数的执行时间统计:
-采用CPU Timer
-采用GPU Timer,也就是CUDA提供的API
-采用NVIDIA Profiler,一种是命令行工具nvprof,另一种是图形化界面的Visual profiler
下面一一介绍。

CPU Timer

CPU Timer可以通过调用系统函数实现,比如在Linux系统上:

#include <sys/time.h>
double cpuSecond() 
{
    struct timeval tp;
    gettimeofday(&tp,NULL);
    return ((double)tp.tv_sec + (double)tp.tv_usec*1.e-6);
}

你可以使用cpuSecond()采用如下的方式来统计kernel函数的执行时间:

double iStart = cpuSecond();
kernel_name<<<grid, block>>>(argument list);
cudaDeviceSynchronize();  //wait for all GPU threads to complete
double iElaps = cpuSecond() - iStart;

由于kernel函数是异步执行的,所以在kernel函数后面统计时间戳前面加上了CUDADeviceSynchronize()以等待kerne函数的完成。为了保险起见,在获取iStart之前也应该调用一下同步函数,这是因为前面可能会有没有执行完成的kernel函数或者是调用了异步的CUDA API。这将导致错误的时间统计。

GPU Timer

CUDA Event 的 API也具有时间统计的功能。cudaEventElapsedTime () 函数可以计算两个事件(event)的间隔时间。返回的结果是以ms为单位,并且其时间分辨率大约为0.5us。其具体的使用方法如下:

cudaEvent_t start, stop;
cudaEventCreate(&start);    
cudaEventCreate(&stop); 
    
cudaEventRecord(start);     
...kernel functions...
cudaEventRecord(stop);               
cudaEventSynchronize(stop);             
cudaEventElapsedTime(&time, start, stop); 
 
cudaEventDestroy(start);                
cudaEventDestroy(stop); 

首先创建两个时间start与stop,然后采用cudaEventRecord记录一个事件,默认情况下是在default Stream (sstream=0)上。在kernel函数执行之后,再重新记录一个事件stop。cudaEventSynchronize()的功能是等待时间stop完成。最后采用cudaEventElapsedTime计算两个时间start与stop之间的时间间隔。

nvprof

采用nvprof可以直接打印每个kernel函数被调用的次数以及执行的时间,另外它还统计了每个CUDA API的调用情况。下面是一个示例:

==9703== Profiling application: ./a.out
==9703== Profiling result:
            Type  Time(%)      Time     Calls       Avg       Min       Max  Name
 GPU activities:  100.00%  111.00ms         3  37.000ms  36.950ms  37.094ms  testKernel(int*)
                    0.00%  2.6560us         3     885ns     736ns  1.1520us  [CUDA memset]
      API calls:   78.69%  418.81ms         2  209.41ms  271.66us  418.54ms  cudaFree
                   13.88%  73.896ms         2  36.948ms  36.943ms  36.953ms  cudaDeviceSynchronize
                    6.97%  37.086ms         1  37.086ms  37.086ms  37.086ms  cudaEventSynchronize
                    0.16%  868.42us         1  868.42us  868.42us  868.42us  cuDeviceTotalMem
                    0.14%  723.05us        96  7.5310us     183ns  329.77us  cuDeviceGetAttribute
                    0.05%  273.57us         1  273.57us  273.57us  273.57us  cudaMalloc
                    0.04%  221.74us         3  73.911us  42.784us  132.13us  cudaMemset
                    0.03%  141.77us         3  47.256us  33.979us  67.575us  cudaLaunchKernel
三种统计方式的比较

很显然,不论是CPU Timer还是GPU Timer我们都需要修改源代码,所以nvprof的优势就是使用更加简单便捷。不需要修改任何代码就可以获得kernel函数的执行时间。当然在源代码中使用Timer更加的灵活,我们可以仅打印我们所关心的代码段的执行时间。所以具体应该使用哪一个,就要看自己的需求了。下面的一段代码展示了CPU Timer与GPU Timer的使用。

#include <stdio.h>
#include <stdlib.h>
#include <cuda.h>
#include <sys/time.h>

//in ms
double cpuMillisecond()
{
    struct timeval tp;
    gettimeofday(&tp,NULL);
    return ((double)tp.tv_sec * 1000 + (double)tp.tv_usec*1.e-3);
}

__global__ void testKernel(int *a)
{
    for(size_t i = 0; i < 1024 * 1024; i++)
    {
        a[0] += a[i];
    }
}

int main() {
    int *d_a;
    int N = 2 << 20; //1M

    cudaFree(0);
    cudaMalloc(&d_a, N * sizeof(int));
    cudaMemset(d_a, 0, N * sizeof(int));

    //warp up
    testKernel<<<1,1>>>(d_a);
    cudaMemset(d_a, 0, N * sizeof(int));

    //need to be added befor cpu timer
    //you can guess what will happen when omitting this synchronization
    cudaDeviceSynchronize();
    //cpuTimer
    double iStart = cpuMillisecond();
    testKernel<<<1,1>>>(d_a);
    cudaDeviceSynchronize();
    double tcpu = cpuMillisecond() - iStart;
    printf("cpuTimer: %.3f ms\n", tcpu);


    //gpuTimer
    float tgpu = 0.f;
    cudaEvent_t start, stop;
    cudaEventCreate(&start);
    cudaEventCreate(&stop);
    cudaMemset(d_a, 0, N * sizeof(int));

    cudaEventRecord(start);
    testKernel<<<1,1>>>(d_a);
    cudaEventRecord(stop);
    cudaEventSynchronize(stop);
    cudaEventElapsedTime(&tgpu, start, stop);
    printf("gpuTimer: %.3f ms\n", tgpu);

    cudaEventDestroy(start);
    cudaEventDestroy(stop);
    cudaFree(d_a);
}

在一台P5000的显卡上执行的结果是:

cpuTimer: 37.241 ms
gpuTimer: 36.999 ms

采用nvprof的统计结果是:

            Type  Time(%)      Time     Calls       Avg       Min       Max  Name
 GPU activities:  100.00%  110.79ms         3  36.931ms  36.773ms  37.081ms  testKernel(int*)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,793评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,567评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,342评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,825评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,814评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,680评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,033评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,687评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,175评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,668评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,775评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,419评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,020评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,206评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,092评论 2 351
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,510评论 2 343