CUDA Event测试时间模板

    // 使用 event 的计算时间
    float time_elapsed = 0;
    cudaEvent_t start, end;
    cudaEventCreate(&start);
    cudaEventCreate(&end);

    cudaEventRecord(start, 0);   // 记录当前时间
    // xxx<<>>(xxx, xxx); // kernel 函数
    cudaEventRecord(end, 0);     // 记录当前时间

    cudaEventSynchronize(start); // 等待 event 完成
    cudaEventSynchronize(end);   // 等待 event 完成
    cudaEventElapsedTime(&time_elapsed, start, end); // 计算时间差

    cudaEventDestroy(start);
    cudaEventDestroy(end);
    printf("执行时间: %f(ms)\n", time_elapsed);

你可能感兴趣的:(CUDA)