jax
Phambinh/skip fused attention navi lds limit
#720

Merged

Phambinh/skip fused attention navi lds limit #720

phambinhfin merged 3 commits into rocm-jaxlib-v0.8.2 from phambinh/skip-fused-attention-navi-lds-limit

Skip fused attention tests on Navi/RDNA GPUs due to 64KB LDS limit

1ca3b021

Skip test_distributed_jax_visible_devices when fewer than 4 GPUs

d522d1b6

phambinhfin requested a review 43 days ago

phambinhfin force pushed from 57aaf2b6 to 94878f22 43 days ago

Ruturaj4 approved these changes on 2026-03-03

Arech8 commented on 2026-03-03

Skip TF32_TF32_F32 dot algorithm test on Navi/RDNA GPUs

171e73c8

i-chaochen commented on 2026-03-03

phambinhfin force pushed from 94878f22 to 171e73c8 43 days ago

phambinhfin enabled auto-merge (squash) 42 days ago

phambinhfin merged 164fc708 into rocm-jaxlib-v0.8.2 42 days ago

Reviewers

Ruturaj4

arsic3

i-chaochen

Arech8

Assignees

No one assigned

Labels

None yet

Milestone

No milestone