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

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

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

 

#define CHECK(status) \
    do\
    {\
        auto ret = (status);\
        if (ret != 0)\
        {\
            std::cerr << "Cuda failure: " << ret << std::endl;\
            abort();\
        }\
    } 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