update submodule ck locally

ROCm · Dec 29, 2024 · 17b2f53 · 17b2f53
1 parent 423b9c2
commit 17b2f53
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/csrc/composable_kernel b/csrc/composable_kernel
diff --git a/csrc/flash_attn_ck/mha_bwd.cpp b/csrc/flash_attn_ck/mha_bwd.cpp
@@ -27,7 +27,7 @@ fmha_bwd_traits get_ck_fmha_bwd_traits(const mask_info &mask,
                            true, // uses_ext_asm
                            true, // is_v3_atomic_fp32
                            false, // is_v3_spec
-                           1}; // how_v3_bf16_cvt 0:RTNE; 1:RTNA; 2:RTZ
+                           2}; // how_v3_bf16_cvt 0:RTNE; 1:RTNA; 2:RTZ
 }
 
 fmha_bwd_args get_ck_fmha_bwd_args(const mask_info &mask,