Skip to content

GitLab

Commit 9f24a7ed authored by Jim's avatar Jim
Browse files

fix: add template defination

parent 7d45045c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py
View file @ 9f24a7ed
......@@ -317,6 +317,9 @@ struct fmha_bwd_dq_dk_dv_v3_traits_
static constexpr bool kIsHDPad = kIsHDPad_;
}};
template <typename fmha_bwd_dq_dk_dv_v3_traits_> struct FmhaBwdV3Name;
template <typename fmha_bwd_dq_dk_dv_v3_traits_> struct FmhaBwdV3Buf;
template <typename fmha_bwd_dq_dk_dv_v3_traits_> struct FmhaBwdV3Ts;
{F_template}
class fmha_bwd_v3_kernel
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment