如何通过宏或SFINAE自动显式实例化多模板参数模板函子

在CUDA环境中，需要为多个模板函子（Functor1、Functor2）与多个数据类型结构体（D1、D2）的所有组合自动生成显式模板实例化代码（如template class Functor1<D1>;），希望通过宏或SFINAE技术实现自动化，替代手动编写重复代码。

方法一：预处理器宏批量生成 #

宏是最直接的批量代码生成方案，适合CUDA这种需要显式实例化的场景，扩展和维护都很简单：

// 定义数据类型列表，新增类型直接追加即可
#define DATA_TYPES \
    D1, \
    D2

// 定义函子列表，新增函子直接追加即可
#define FUNCTORS \
    Functor1, \
    Functor2

// 辅助宏：为单个函子生成某一数据类型的实例化
#define INSTANTIATE_FUNCTOR_ONE(FUNCTOR, DT) \
    template class FUNCTOR<DT>;

// 辅助宏：为单个函子生成所有数据类型的实例化
#define INSTANTIATE_FUNCTOR_ALL_DT(FUNCTOR) \
    INSTANTIATE_FUNCTOR_ONE(FUNCTOR, D1) \
    INSTANTIATE_FUNCTOR_ONE(FUNCTOR, D2)

// 生成所有函子+数据类型组合的实例化
#define INSTANTIATE_ALL_COMBOS() \
    INSTANTIATE_FUNCTOR_ALL_DT(Functor1) \
    INSTANTIATE_FUNCTOR_ALL_DT(Functor2)

// 调用宏生成代码
INSTANTIATE_ALL_COMBOS()

// 清理宏，避免污染全局命名空间
#undef DATA_TYPES
#undef FUNCTORS
#undef INSTANTIATE_FUNCTOR_ONE
#undef INSTANTIATE_FUNCTOR_ALL_DT
#undef INSTANTIATE_ALL_COMBOS

如果后续需要新增数据类型或函子，只需要修改DATA_TYPES和FUNCTORS两个宏，无需改动实例化逻辑。

方法二：SFINAE+模板元编程实现 #

如果需要更灵活的类型遍历（比如条件过滤实例化），可以用C++模板元编程结合SFINAE实现：

#include <type_traits>

// 定义类型列表模板
template<typename... Ts>
struct TypeList {};

// 定义需要实例化的类型集合
using TargetDataTypes = TypeList<D1, D2>;
using TargetFunctors = TypeList<Functor1, Functor2>;

// 为单个函子实例化所有目标数据类型
template<typename Functor, typename DataTypeList>
struct InstantiateFunctor;

template<typename Functor, typename... DTs>
struct InstantiateFunctor<Functor, TypeList<DTs...>> {
    // 用折叠表达式触发每个类型的显式实例化
    static void trigger() {
        (void)std::initializer_list<int>{
            (template class Functor<DTs>, 0)...
        };
    }
};

// 遍历所有函子，触发各自的实例化逻辑
template<typename FunctorList, typename DataTypeList>
struct InstantiateAll;

template<typename... Fs, typename DataTypeList>
struct InstantiateAll<TypeList<Fs...>, DataTypeList> {
    static void trigger() {
        (void)std::initializer_list<int>{
            (InstantiateFunctor<Fs, DataTypeList>::trigger(), 0)...
        };
    }
};

// 触发所有组合的实例化（用匿名函数确保编译期执行）
static auto __force_instantiate = [](){
    InstantiateAll<TargetFunctors, TargetDataTypes>::trigger();
    return 0;
}();

注意：在CUDA环境中，需确保模板函子的__device__/__global__成员函数已完整定义，显式实例化会将这些函数编译进目标文件，避免链接阶段报错。

内容的提问来源于stack exchange，提问作者Eugene W.