c++ - 考虑到模板参数改变 if 语句条件

Question

我有一个带有模板参数的全局内核函数:

template<int ARG> __global__ void kernel(array[]) {
    int threadID = blockDim.x*blockIdx.x + threadIdx.x;
    if(...) {...}
}

考虑到模板参数，函数的行为和特殊的 if 语句条件略有不同，但主体保持不变。让我们说:
ARG == 0 if 语句如下:if(expr1){body}
ARG == 1 if 语句如下:if(expr2){body}
ARG == 2 if 语句如下:if(expr1 && expr2){body}

我的问题是(在可读性和性能方面)提供此功能的最佳方式是什么？

编辑:表达式 expr1 和 expr2 是对 __device__ boolean 函数的调用，例如fnc1(array[threadID]) 和 fnc2(array[threadID])。

Answer 1

直接的方法是蛮力:

if ((ARG != 1 || expr1) && (ARG != 0 || expr2)) ...

由于 ARG 在编译时已知，因此编译器将在此处生成良好的代码。