Skip to content

Commit 95499a6

Browse files
alugoreydnikolaev-amd
authored andcommitted
Update CK for gfx950 (#49)
(cherry picked from commit 79922930ae070e81e9a2aadf9b56b66a8d8c95b1)
1 parent 9f72899 commit 95499a6

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

aten/src/ATen/native/transformers/hip/flash_attn/ck/fmha_bwd.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -453,4 +453,5 @@ struct fmha_bwd_traits
453453
bool is_deterministic;
454454
// TODO: padding check is inside this api
455455
};
456+
template <int Version = 2>
456457
float fmha_bwd(fmha_bwd_traits, fmha_bwd_args, const ck_tile::stream_config&);

third_party/composable_kernel

Submodule composable_kernel updated 836 files

0 commit comments

Comments
 (0)