qazalandGitHub 939f28d571 fused qkv rope custom kernel (#17021) ...

* work

* fused qkv_norm

* work

* speed

* not that yet

* test cleanup

* just clone

* remove .realize()

* cleanup tests

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

..

include

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

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

fa_bwd_causal.cpp

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

2026-07-14 00:27:00 +09: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: support D=64 and attn sink (#16935)

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

fa.py

fused qkv rope custom kernel (#17021)

2026-07-15 01:08:42 +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