C++：混淆可变参数模板导致 "candidate function not viable" 错误

Question

我试图将一个 CUDA 项目移植到广泛使用 C++ 模板的 ROCm 平台。在此过程中，出现以下编译错误

/root/warp-ctc/include/detail/gpu_ctc.h:381:5: error: no matching function for call to 'hipLaunchKernelGGL'
    hipLaunchKernelGGL((prepare_stable_SM_kernel<ProbT, VT>), dim3(grid_size), dim3(NT), 0, stream_, ctc_helper::identity<ProbT>(), probs_,
    ^~~~~~~~~~~~~~~~~~
.....
.....
/opt/rocm/hip/include/hip/hcc_detail/functional_grid_launch.hpp:138:13: note: candidate function [with Args = <ctc_helper::identity<float, float>, float *,
      float *, int, int>, F = void (*)(ctc_helper::identity<float, float>, float *, float *, int, int)] not viable: no overload of 'prepare_stable_SM_kernel'
      matching 'void (*)(ctc_helper::identity<float, float>, float *, float *, int, int)' for 1st argument
inline void hipLaunchKernelGGL(F kernel, const dim3& numBlocks, const dim3& dimBlocks,

对于following piece of C++ code

hipLaunchKernelGGL((prepare_stable_SM_kernel<ProbT, VT>),
                   dim3(grid_size),
                   dim3(NT),
                   0,
                   stream_,
                   ctc_helper::identity<ProbT>(),
                   probs_,
                   denoms_,
                   out_dim_,
                   num_elements);

其中 hipLaunchKernelGGL 是 defined as

template <typename... Args, typename F = void (*)(Args...)>
inline void hipLaunchKernelGGL(F kernel,
                               const dim3& numBlocks,
                               const dim3& dimBlocks,
                               std::uint32_t sharedMemBytes,
                               hipStream_t stream,
                               Args... args) {

// ...
// ...
}

和prepare_stable_SM_kernel是defined as

template <typename ProbT, int VT = 1, typename Op>
__global__ void prepare_stable_SM_kernel(Op f, ProbT* probs,
                                         const ProbT* const col_max,
                                         int alphabet_size,
                                         int count) {
// ...
}

任何人请提供一些提示来解决这个问题，帮助我。

Answer 1

此错误已通过向 prepare_stable_SM_kernel 函数

添加第三个参数 ctc_helper::identity<ProbT> 修复

参考 https://github.com/harish2704/warp-ctc/commit/c7ec45febc9c0077ffa35932b3d11a05daf8bf7c#diff-1e372cfc8edab301b6c284e9db8dbf68

C++：混淆可变参数模板导致 "candidate function not viable" 错误

C++: confusing variadic template causing "candidate function not viable" error

c++

variadic-templates

amd-rocm