Replace TileFmhaBwdTraits by TileFmhaTraits

ROCm · Oct 15, 2024 · a15e559 · a15e559
1 parent 723c420
commit a15e559
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/third_party/composable_kernel_tiled b/third_party/composable_kernel_tiled
diff --git a/xformers/csrc/attention/hip_fmha/ck_tiled_fmha_batched_backward.h b/xformers/csrc/attention/hip_fmha/ck_tiled_fmha_batched_backward.h
@@ -115,7 +115,7 @@ struct batched_backward_causalmask_bias_dropout_dispatch {
 
         BOOL_SWITCH_2(
             pad_headdim_q, kPadHeadDimQ, pad_headdim_v, kPadHeadDimV, [&] {
-              using FmhaBwdTraits_ = ck_tile::TileFmhaBwdTraits<
+              using FmhaBwdTraits_ = ck_tile::TileFmhaTraits<
                   kPadSeqLenQ,
                   kPadSeqLenK,
                   kPadHeadDimQ,

diff --git a/xformers/csrc/attention/hip_fmha/ck_tiled_fmha_grouped_backward.h b/xformers/csrc/attention/hip_fmha/ck_tiled_fmha_grouped_backward.h
@@ -113,7 +113,7 @@ struct grouped_backward_causalmask_bias_dropout_dispatch {
 
         BOOL_SWITCH_2(
             pad_headdim_q, kPadHeadDimQ, pad_headdim_v, kPadHeadDimV, [&] {
-              using FmhaBwdTraits_ = ck_tile::TileFmhaBwdTraits<
+              using FmhaBwdTraits_ = ck_tile::TileFmhaTraits<
                   kPadSeqLenQ,
                   kPadSeqLenK,
                   kPadHeadDimQ,