From 9f24a7ed410e18a283a1e7680bab89287478a34b Mon Sep 17 00:00:00 2001 From: Jim Date: Wed, 12 Feb 2025 17:34:55 +0800 Subject: [PATCH] fix: add template defination --- example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py b/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py index 0b9bff426a..b867bcc720 100644 --- a/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py +++ b/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py @@ -317,6 +317,9 @@ struct fmha_bwd_dq_dk_dv_v3_traits_ static constexpr bool kIsHDPad = kIsHDPad_; }}; +template struct FmhaBwdV3Name; +template struct FmhaBwdV3Buf; +template struct FmhaBwdV3Ts; {F_template} class fmha_bwd_v3_kernel