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