jax
Skip test_get_vmap for slice-only case with mismatched batch dims
#636

Merged

Skip test_get_vmap for slice-only case with mismatched batch dims #636

phambinhfin merged 2 commits into rocm-jaxlib-v0.8.0 from phambinh/skip-get-vmap-slice-only-bug

Skip test_get_vmap for slice-only case with mismatched batch dims

e65ca2e9

phambinhfin requested a review 35 days ago

Skip test_get_vmap for known _get_vmap slice-only bug

21a3d50d

gulsumgudukbay approved these changes on 2026-01-22

phambinhfin merged f8dc4b1f into rocm-jaxlib-v0.8.0 35 days ago

Reviewers

gulsumgudukbay

Assignees

No one assigned

Labels

None yet

Milestone

No milestone