C++宏的一些较复杂的用法,反斜杠,do {}while(0)

最近学习CUDA编程,见以下代码:https://github.com/wang-xinyu/tensorrtx/blob/master/lenet/lenet.cpp

其中经常出现一个检查内存分配是否政策的宏函数CHECK:

 

 1 #define CHECK(status) \
 2     do\
 3     {\
 4         auto ret = (status);\
 5         if (ret != 0)\
 6         {\
 7             std::cerr << "Cuda failure: " << ret << std::endl;\
 8             abort();\
 9         }\
10     } while (0)

反斜杠将本行与下一行连接起来,使宏语句维持在一行中。

宏中使用do{}while(0)包裹两条及以上的语句,使之成为一个整体,这常常用在判断语句if之后的宏中(虽然此处CHECK的调用上下文中并没有if)。由于把宏函数引入代码会自动加上分号,如果在宏函数定义中使用 { } 大括号包裹,在if ... else ...的结构中会多出来一个"}"后的分号。使用do{….}while(0) 把它包裹起来,成为一个独立的语法单元,从而不会与上下文发生混淆。

而在cuda的内存分配函数中,统一返回的是cudaError枚举类型,其中:

返回值0表示没有错误。

 

参考:

https://www.cnblogs.com/anker/p/3418792.html

https://stackoverflow.com/questions/257418/do-while-0-what-is-it-good-for

https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__TYPES.html#group__CUDART__TYPES_1g3f51e3575c2178246db0a94a430e0038

https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__STREAM.html

posted @ 2021-06-01 15:57  略略略——  阅读(726)  评论(0编辑  收藏  举报