[0.6.0-UT] Fix Test Shard#519

Open

AratiGanesh wants to merge 1 commit intorocm-jaxlib-v0.6.0from

Commits on Jul 29, 2025

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