編寫一個C宏以在CUDA內核中使用

Question

我在代碼中具有以下結構，並且已經使用了很多次。所以，爲了提高代碼的可讀性和減少行數，我真的需要使用宏來代替。其中我期待編寫一個宏爲它的部分如下：

通過任意計算X

#define _UNROLL_FACTOR_volIntGrad 32 
    int jj = 0; 
    for (; jj < (ngbSize - 32); jj += 32) { 
     int j = offset + jj; 
#pragma unroll 
     for (int k = 0; k < 32; k++){ 
     ... 
     arbitrary calculation 1 (depends on k) 
     ... 
     } 
     ... 
     arbitrary calculation 2 
     ... 
    } 

    for (; jj < (ngbSize - (_UNROLL_FACTOR_volIntGrad/2)); jj+= (_UNROLL_FACTOR_volIntGrad/2)){ 
     int j = offset + jj; 
#pragma unroll 
     for (int k = 0; k < 16; k++){ 
     ... 
     arbitrary calculation 1 (depends on k) 
     ... 
     } 
     ... 
     arbitrary calculation 2 
     ... 
    } 

    for (; jj < (ngbSize - (_UNROLL_FACTOR_volIntGrad/4)); jj+= (_UNROLL_FACTOR_volIntGrad/4)){ 
     int j = offset + jj; 
#pragma unroll 
     for (int k = 0; k < 8; k++){ 
     ... 
     arbitrary calculation 1 (depends on k) 
     ... 
     } 
     ... 
     arbitrary calculation 2 
     ... 
    } 

    for (; jj < (ngbSize - (_UNROLL_FACTOR_volIntGrad/8)); jj+= (_UNROLL_FACTOR_volIntGrad/8)){ 
     int j = offset + jj; 

#pragma unroll 
     for (int k = 0; k < 4; k++){ 
     ... 
     arbitrary calculation 1 (depends on k) 
     ... 
     } 
     ... 
     arbitrary calculation 2 
     ... 
    } 

    for (; jj < (ngbSize - (_UNROLL_FACTOR_volIntGrad/16)); jj+= (_UNROLL_FACTOR_volIntGrad/16)){ 
     int j = offset + jj; 
#pragma unroll 
     for (int k = 0; k < 2; k++){ 
     ... 
     arbitrary calculation 1 (depends on k) 
     ... 
     } 
     ... 
     arbitrary calculation 2 
     ... 
    } 
    for (; jj < ngbSize; jj++){ 
     int j = offset + jj; 
     ... 
     arbitrary calculation 3 
     ... 
    } 
} 

，我的意思是一組計算的是獨立的宏觀和功能不同的功能。有誰知道如何編寫這個宏，以減少上述結構的大小？比如像下面這樣：

__MACRO 
    arbitrary calculation 1 
    arbitrary calculation 2 
    arbitrary calculation 3 
__END 

Answer 1

在C++中，它主要是不贊成使用宏比包括警衛和平臺依賴性條件編譯的任何其他。最好的辦法是創建一個靜態常量，它將在內部鏈接並具有單一維護點。你可以把它放在你的文件的頂部。

如果您使用的是C++ 11，那麼您可以使用constexpr來實現您正在嘗試的操作。編譯器會知道你的語句有一個類型，而不僅僅是一個文本替換，這實際上是C風格的宏所做的。

constexpr的目的是創建一個可以在編譯時計算的不可變對象，有點像靜態常量。但是關於它們的好處是你可以用它們創建靜態函數，所以在你進行依賴於其他計算的情況下會很有用。

查看使用情況constexpr這裏： When should you use constexpr capability in C++11?