Skip to content

Conversation

@gulsumgudukbay
Copy link

The reshape call has a numerical bug where the target reshape size is always higher than array size if num_devices is not a nonnegative power of two. To fix this, this commit adds padding.

… always higher than array size if num_devices is not a nonnegative power of two. To fix this, this commit adds padding
@gulsumgudukbay gulsumgudukbay requested a review from a team as a code owner July 16, 2025 19:25
@gulsumgudukbay gulsumgudukbay merged commit 3d6b521 into rocm-jaxlib-v0.6.0 Jul 16, 2025
@gulsumgudukbay gulsumgudukbay deleted the fix_ann_test_test_pmap branch July 16, 2025 19:41
@gulsumgudukbay gulsumgudukbay changed the title Fix ann_test::test_pmap [0.6.0-UT] Fix ann_test::test_pmap Jul 29, 2025
@gulsumgudukbay gulsumgudukbay restored the fix_ann_test_test_pmap branch September 8, 2025 21:54
@gulsumgudukbay gulsumgudukbay deleted the fix_ann_test_test_pmap branch September 8, 2025 22:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants