danyao12 60178ac970 hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

..

bwd_bf16_a16_rtna_pddv.cpp

fix hd72~120 memory fault

2025-02-08 13:46:39 +08:00

bwd_bf16_a16_rtna.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_a16_rtne_pddv.cpp

fix hd72~120 memory fault

2025-02-08 13:46:39 +08:00

bwd_bf16_a16_rtne.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_a16_rtz_pddv.cpp

fix hd72~120 memory fault

2025-02-08 13:46:39 +08:00

bwd_bf16_a16_rtz.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_a32_rtna_psskddv.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_bf16_a32_rtna.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_a32_rtne_psskddv.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_bf16_a32_rtne.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_a32_rtz_psskddv.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_bf16_a32_rtz.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_causal_a16_rtna_pddv.cpp

fix hd72~120 memory fault

2025-02-08 13:46:39 +08:00

bwd_bf16_causal_a16_rtna.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_causal_a16_rtne_pddv.cpp

fix hd72~120 memory fault

2025-02-08 13:46:39 +08:00

bwd_bf16_causal_a16_rtne.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_causal_a16_rtz_pddv.cpp

fix hd72~120 memory fault

2025-02-08 13:46:39 +08:00

bwd_bf16_causal_a16_rtz.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_causal_a32_rtna_psskddv.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_bf16_causal_a32_rtna.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_causal_a32_rtne_psskddv.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_bf16_causal_a32_rtne.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_bf16_causal_a32_rtz_psskddv.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_bf16_causal_a32_rtz.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_fp16_a16_pddv.cpp

fix hd72~120 memory fault

2025-02-08 13:46:39 +08:00

bwd_fp16_a16.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_fp16_a32_psskddv.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_fp16_a32.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_fp16_causal_a16_pddv.cpp

fix hd72~120 memory fault

2025-02-08 13:46:39 +08:00

bwd_fp16_causal_a16.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_fp16_causal_a32_psskddv.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_fp16_causal_a32.cpp

separate hd pad/unpad kernels

2025-01-24 13:55:02 +08:00

bwd_hd64_bf16_a16_rtna.cpp

fix hd64 seqlen64 memory fault

2025-01-13 12:14:51 +08:00

bwd_hd64_bf16_a16_rtne.cpp

fix hd64 seqlen64 memory fault

2025-01-13 12:14:51 +08:00

bwd_hd64_bf16_a16_rtz.cpp

fix hd64 seqlen64 memory fault

2025-01-13 12:14:51 +08:00

bwd_hd64_bf16_a32_rtna_pssk.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_hd64_bf16_a32_rtne_pssk.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_hd64_bf16_a32_rtz_pssk.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_hd64_bf16_causal_a16_rtna.cpp

fix hd64 seqlen64 memory fault

2025-01-13 12:14:51 +08:00

bwd_hd64_bf16_causal_a16_rtne.cpp

fix hd64 seqlen64 memory fault

2025-01-13 12:14:51 +08:00

bwd_hd64_bf16_causal_a16_rtz.cpp

fix hd64 seqlen64 memory fault

2025-01-13 12:14:51 +08:00

bwd_hd64_bf16_causal_a32_rtna_pssk.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_hd64_bf16_causal_a32_rtne_pssk.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_hd64_bf16_causal_a32_rtz_pssk.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_hd64_fp16_a16.cpp

add hd64 fp16 kernels

2025-02-05 16:24:21 +08:00

bwd_hd64_fp16_a32_pssk.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

bwd_hd64_fp16_causal_a16.cpp

add hd64 fp16 kernels

2025-02-05 16:24:21 +08:00

bwd_hd64_fp16_causal_a32_pssk.cpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00

fmha_hsaco.hpp

hd64~128(x8) fp16/bf16 a32 w/cas_kb arbitrary seqlen_q/seqlen_k/strides

2025-02-28 15:17:27 +08:00