jax
Enable test_dot_general_vmap tests on ROCm
#618
Merged

Enable test_dot_general_vmap tests on ROCm #618

phambinhfin
phambinhfin Enable test_dot_general_vmap tests on ROCm
4d00cbdb
phambinhfin phambinhfin requested a review 12 hours ago
gulsumgudukbay
gulsumgudukbay approved these changes on 2026-01-20
phambinhfin phambinhfin added open-upstream
phambinhfin phambinhfin added cherry-pick-candidate
gulsumgudukbay gulsumgudukbay merged ea794c2c into rocm-jaxlib-v0.8.0 4 hours ago
gulsumgudukbay gulsumgudukbay deleted the phambinh/enable-vmap-dot-tests-rocm branch 4 hours ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone