Correct the dtype checking logics

example/ck_tile/01_fmha/codegen/ops/fmha_fwd_splitkv.py

@@ -282,7 +282,7 @@ FMHA_FWD_SPLITKV_API_INNER_DISPATCH="""            {F_if}((t.is_group_mode == {F
                 if (1 < a.num_splits) {{
                     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}>;
                     if (t.has_lse) {{
-                        if constexpr (std::is_same_v<{F_dtype}, ck_tile::fp8_t>) {{
+                        if constexpr (std::is_same_v<{F_dtype}, FmhaFwdFp8>) {{
                             return -1;
                         }} else {{
                             using traits2_ = fmha_fwd_splitkv_combine_traits_<{F_hdim}, {F_dtype}, {F_mode}, {F_bm0}/2, {F_bn1}/2, true, {F_squant}, {F_spad}, {F_dvpad}>;