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

Question

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

然而，我已经使用-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接口函数在运行时获取所需的值。