PR #519 [0.6.0-UT] Fix Test Shard

[0.6.0-UT] Fix Test Shard #519

AratiGanesh wants to merge 1 commit into rocm-jaxlib-v0.6.0 from fix_test_shard

Fix to trim the input to be divisible by num gpus in test shard UT

e7e847ad

AratiGanesh requested a review 144 days ago

AratiGanesh changed the title ~~[0.6.0-UT] Fix test shard~~ [0.6.0-UT] Fix Test Shard 143 days ago

gulsumgudukbay requested changes on 2025-09-10

Reviewers

gulsumgudukbay

Assignees

No one assigned

Labels

None yet

Milestone

No milestone