如何限制Numba(CUDA)中每个线程使用的寄存器数量

2024-04-19 12:29:21 发布

您现在位置:Python中文网/ 问答频道 /正文

正如标题所说,我想知道在启动内核时是否有办法限制每个线程使用的寄存器数量。 我在每个线程上执行大量的计算,所以使用的寄存器数量太多,占用率很低。我想尝试减少使用的寄存器的数量,以改善并行线程的执行,也许是以更多的内存访问为代价的。你知道吗

我寻找答案,但没有找到解决办法。我认为可以通过CUDA工具链设置线程使用的最大寄存器数,但是在使用NUBA时也可以吗?你知道吗

编辑:也可以强制在多处理器中执行最少数量的块,以迫使编译器减少已用寄存器的数量。你知道吗


Tags: 工具内存答案编辑标题数量线程内核
1条回答
网友
1楼 · 发布于 2024-04-19 12:29:21

据我所知,numba提供的cuda.jit工具不允许将参数传递给CUDA汇编程序,这将允许控制寄存器分配,就像本机CUDA工具链一样。你知道吗

所以我不认为有一种方法可以满足你的要求。你知道吗

相关问题 更多 >