cuda - 在内核调用中使用断言

Question

是否有方便的方法在设备模式下的内核调用中使用断言？

Answer 1

CUDA 现在具有原生断言函数。使用assert(...)。如果其参数为零，它将停止内核执行并返回错误。 (如果在 CUDA 调试中，则触发断点。)

确保包含“assert.h”。此外，这需要计算能力 2.x 或更高版本，并且在 MacOS 上不受支持。有关更多详细信息，请参阅 CUDA C 编程指南，第 B.16 节。

编程指南还包括以下示例:

#include <assert.h>
__global__ void testAssert(void)
{
   int is_one = 1;
   int should_be_one = 0;
   // This will have no effect
   assert(is_one);
   // This will halt kernel execution
   assert(should_be_one);
}
int main(int argc, char* argv[])
{
   testAssert<<<1,1>>>();
   cudaDeviceSynchronize();
   return 0;
}