jax
aiter mha kernels (ASM+CK) integration
#747
Merged
Overview
Commits
8
Changes
View On GitHub

aiter mha kernels (ASM+CK) integration #747

zahiqbal merged 8 commits into aiter-mha from aiter-mha-integration-draft-1
zahiqbal
zahiqbal zahiqbal requested a review from Ruturaj4 Ruturaj4 167 days ago
zahiqbal zahiqbal requested a review from gulsumgudukbay gulsumgudukbay 167 days ago
zahiqbal zahiqbal requested a review from JehandadKhan JehandadKhan 167 days ago
zahiqbal zahiqbal requested a review from phambinhfin phambinhfin 167 days ago
phambinhfin
zahiqbal
zahiqbal aiter mha kernels (ASM+CK) integration
a6041a3b
zahiqbal zahiqbal force pushed from 0df02f78 to a6041a3b 167 days ago
i-chaochen
i-chaochen commented on 2026-04-08
wangye805
wangye805 commented on 2026-04-08
zahiqbal zahiqbal requested a review from Arech8 Arech8 161 days ago
Arech8
Arech8 commented on 2026-04-13
Arech8
Arech8 commented on 2026-04-13
Arech8
Arech8 commented on 2026-04-13
zahiqbal code cleanup and formatting after code review
720431d3
zahiqbal zahiqbal force pushed from 3404a0dc to 720431d3 157 days ago
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 commented on 2026-04-20
Arech8
Arech8 requested changes on 2026-04-20
zahiqbal zahiqbal requested a review from mminutoli mminutoli 153 days ago
gulsumgudukbay
gulsumgudukbay requested changes on 2026-04-21
zahiqbal handling aiter-mha libs in wheel at rocm/jax fork
f461fcf3
zahiqbal zahiqbal force pushed from 6f80b73b to f461fcf3 153 days ago
zahiqbal upgraded aiter libs
121df61e
zahiqbal corrected mha asm_v3 vs ck kernel calls
b616cf94
Arech8
Arech8 commented on 2026-04-23
Arech8
Arech8 commented on 2026-04-23
zahiqbal zahiqbal force pushed from ab93aac3 to bab84f19 151 days ago
zahiqbal zahiqbal force pushed from bab84f19 to ddbe075e 151 days ago
zahiqbal zahiqbal force pushed from ddbe075e to 14c65f4d 151 days ago
zahiqbal zahiqbal force pushed from 14c65f4d to a9ff0f58 151 days ago
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Ruturaj4
Ruturaj4 approved these changes on 2026-04-24
zahiqbal code review updates
ca02de70
zahiqbal zahiqbal force pushed from a9ff0f58 to ca02de70 150 days ago
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 commented on 2026-04-24
Arech8
Arech8 approved these changes on 2026-04-24
zahiqbal fix correctness, concurrency, multi-GPU and FFI safety in aiter mha
1d827fcb
zahiqbal zahiqbal force pushed from 8ea0555b to 1d827fcb 150 days ago
zahiqbal aiter libs alternative path to local dir, --repo_env=LOCAL_AITER_PATH…
46d314d4
zahiqbal zahiqbal merged 23b4a327 into aiter-mha 148 days ago
i-chaochen
i-chaochen
i-chaochen commented on 2026-05-05

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum