Skip to content

GitLab

Commit ed634ea4 authored by Po Yen Chen's avatar Po Yen Chen
Browse files

Fix wrong trait template arg

parent 47e523ef
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
example/ck_tile/01_fmha/codegen/ops/fmha_fwd_splitkv.py
View file @ ed634ea4
...@@ -289,7 +289,7 @@ FMHA_FWD_SPLITKV_API_INNER_DISPATCH=""" {F_if}((t.is_group_mode == {F ...@@ -289,7 +289,7 @@ FMHA_FWD_SPLITKV_API_INNER_DISPATCH=""" {F_if}((t.is_group_mode == {F
return fmha_fwd_splitkv_<traits_>(s, a); return fmha_fwd_splitkv_<traits_>(s, a);
}} else {{ }} else {{
using traits_ = fmha_fwd_splitkv_traits_<{F_hdim}, {F_dtype}, {F_mode}, {F_bm0}, {F_bn0}, {F_bk0}, {F_bn1}, {F_bk1}, {F_bk0max}, {F_vlayout}, {F_pipeline_enum}, {F_mask}, {F_bias}, true, {F_squant}, {F_pagedkv}, {F_spad}, {F_skpad}, {F_dpad}, {F_dvpad}>; using traits_ = fmha_fwd_splitkv_traits_<{F_hdim}, {F_dtype}, {F_mode}, {F_bm0}, {F_bn0}, {F_bk0}, {F_bn1}, {F_bk1}, {F_bk0max}, {F_vlayout}, {F_pipeline_enum}, {F_mask}, {F_bias}, false, {F_squant}, {F_pagedkv}, {F_spad}, {F_skpad}, {F_dpad}, {F_dvpad}>;
return fmha_fwd_splitkv_<traits_>(s, a); return fmha_fwd_splitkv_<traits_>(s, a);
}} }}
......
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