ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-13 02:27:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

enable swa

Browse Source
This commit is contained in:
Jim
2025-04-01 10:31:34 +08:00
parent 173ed584e0
commit aeb31a27af
4 changed files with 3286 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

659
example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py
View File

File diff suppressed because it is too large Load Diff

BIN
example/ck_tile/01_fmha/hsaco/bwd_bf16_swa_a32_rtne_psskddv.co Executable file
View File

Binary file not shown.

2855
example/ck_tile/01_fmha/hsaco/bwd_bf16_swa_a32_rtne_psskddv.cpp Normal file
View File

File diff suppressed because it is too large Load Diff

1
example/ck_tile/01_fmha/hsaco/fmha_hsaco.hpp
View File

@@ -31,6 +31,7 @@ extern unsigned char bwd_bf16_causal_a16_rtz_pddv[];
extern unsigned char bwd_bf16_causal_a32_rtna_psskddv[];
extern unsigned char bwd_bf16_causal_a32_rtne_psskddv[];
extern unsigned char bwd_bf16_causal_a32_rtz_psskddv[];
extern unsigned char bwd_bf16_swa_a32_rtne_psskddv[];
extern unsigned char bwd_fp16_a16_pddv[];
extern unsigned char bwd_fp16_a32_psskddv[];
extern unsigned char bwd_fp16_causal_a16_pddv[];