Logo
Explore Help
Register Sign In
ViperEkura/AstrAI
Watch 1
Star 0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
6adc221c104c7808ffa7da21e14a0e86ce64f9dd
AstrAI/csrc/kernels
T
History
ViperEkura 6adc221c10 refactor: extract shared MMA utils into gqa_mma_utils.cuh
- Move mma16816, ld2, pk2, pkb, ldmatrix_x4/x2/x2_trans to shared header
- gqa_prefill_attn_mma.cuh and gqa_decode_attn_mma.cuh both include it
2026-07-07 23:01:15 +08:00
..
gqa_common.cuh
fix: correct prefill mask index, unify GQA kernel interface
2026-07-06 17:21:23 +08:00
gqa_decode_attn_mma.cuh
refactor: extract shared MMA utils into gqa_mma_utils.cuh
2026-07-07 23:01:15 +08:00
gqa_decode_attn.cu
feat: MMA head-packing decode kernel with scalar fallback dispatch
2026-07-07 22:56:02 +08:00
gqa_decode_attn.cuh
fix: correct prefill mask index, unify GQA kernel interface
2026-07-06 17:21:23 +08:00
gqa_mma_utils.cuh
refactor: extract shared MMA utils into gqa_mma_utils.cuh
2026-07-07 23:01:15 +08:00
gqa_prefill_attn_mma.cuh
refactor: extract shared MMA utils into gqa_mma_utils.cuh
2026-07-07 23:01:15 +08:00
gqa_prefill_attn.cu
feat: ldmatrix + smem padding for mma prefill kernel
2026-07-06 20:55:22 +08:00
gqa_prefill_attn.cuh
feat: tensor-core mma prefill with build-time dispatch
2026-07-06 20:33:24 +08:00
Powered by Gitea Version: 1.27.2 Page: 26ms Template: 3ms
Auto
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API