jax
Fix v_pages quantization and adjust test params for ROCm compatibility
#559

Merged

Fix v_pages quantization and adjust test params for ROCm compatibility #559

mminutoli merged 2 commits into rocm-jaxlib-v0.7.1 from phambinh/fix_gpu_paged_testcases

Fix v_pages quantization and adjust test params for ROCm compatibility

52652684

phambinhfin requested a review from

mminutoli 34 days ago

mminutoli commented on 2025-11-19

Adatption input sizes based ROCm hardware for gpu_paged_attention_tes…

36ee5641

phambinhfin force pushed from 951dc3ce to 36ee5641 34 days ago

mminutoli approved these changes on 2025-11-19

mminutoli merged 00e65c32 into rocm-jaxlib-v0.7.1 34 days ago

Reviewers

mminutoli

Assignees

No one assigned

Labels

None yet

Milestone

No milestone