如何在 Linux 中读取 CUDA .cubin 二进制文件？

Question

我正在尝试获取 CUDA 程序中每个内核的每个线程的寄存器数量。这将帮助我使用 CUDA 占用计算器 (http://developer.download.nvidia.com/compute/cuda/CUDA_Occupancy_calculator.xls) 来确定我的程序的 GPU 的最高占用率。

但是，我已经使用 -cubin 标志生成了 .cubin 文件，但无法按照 NVIDIA (http://forums.nvidia.com/index.php?showtopic=31279) 的建议在 vim/othertexteditors 中读取它。有谁知道如何阅读它吗？

谢谢

Answer 1

最简单的解决方案是将 -Xptxas -v 传递给 nvcc ，如下所示

$ nvcc -Xptxas -v foo.cu
ptxas info    : Compiling entry function '_Z9my_kernelPfS_f' for 'sm_10'
ptxas info    : Used 2 registers, 20+16 bytes smem

或者，您可以使用 cudaFuncGetAttributes API 函数，用于在运行时获取必要的值。