jax
Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling
#725
Open
Open
Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling #725
AratiGanesh
wants to merge
1 commit into
rocm-jaxlib-v0.9.0
from
araganes/skip-pmap-sharding-error
Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling
7c8b2e59
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub