wozeparrotandGitHub 1fd6b1035f fa: swa support (#17367)

2026-08-06 08:07:30 -07:00

..

include

fa: swa support (#17367)

2026-08-06 08:07:30 -07:00

fa_bwd_causal.cpp

fa: swa support (#17367)

2026-08-06 08:07:30 -07:00

fa_bwd_post.cpp

fa: support D=64 and attn sink (#16935)

2026-07-08 20:03:47 -07:00

fa_bwd_pre.cpp

fa: support D=64 and attn sink (#16935)

2026-07-08 20:03:47 -07:00

fa_fwd_causal.cpp

fa: swa support (#17367)

2026-08-06 08:07:30 -07:00

fa.py

test MXFP4 llama without hipcc (#17435)

2026-08-06 17:31:40 +09:00

fused_qkv_rope_bwd.cpp

fused qkv rope custom kernel (#17021)

2026-07-15 01:08:42 +09:00

fused_qkv_rope.cpp

fused qkv rope custom kernel (#17021)

2026-07-15 01:08:42 +09:00

gemm_bf16_atb.cpp

llama: a_bT and aT_b bf16 gemms (#16487)

2026-06-04 23:30:21 +09:00

gemm_bf16.cpp

llama: fix bf16 gemm oob (#16603)

2026-06-12 19:43:05 -07:00

gemm_fp8_atb.cpp

llama: gemm/fa backward speedups (gpt 5.6) (#17007)

2026-07-14 00:27:00 +09:00

gemm_fp8.cpp

llama: gemm/fa backward speedups (gpt 5.6) (#17007)

2026-07-14 00:27:00 +09:00

gemm_mxfp8.cpp

gemm: fix mxfp8 on more shapes (#16677)

2026-06-19 13:28:53 -07:00

grouped_mxfp8_gemm.cpp

gptoss: moe gemm kernels (#17178)

2026-07-24 07:25:15 -07:00

grouped_mxfp8_wgrad.cpp

gptoss: moe gemm kernels (#17178)

2026-07-24 07:25:15 -07:00