PR #581 Fix/pallas tests shared memory

Fix/pallas tests shared memory #581

AratiGanesh wants to merge 2 commits into rocm-jaxlib-v0.8.0 from fix/pallas-tests-shared-memory

Add rocm test requirements file

4c770679

- Fix `gpu_ops_test.py`: Implement block size reduction code. Dynamic…

79beef01

AratiGanesh requested a review 6 days ago

AratiGanesh changed the base branch from main to rocm-jaxlib-v0.8.0 6 days ago

AratiGanesh requested a review from

mminutoli 6 days ago

AratiGanesh removed review request 6 days ago

mminutoli requested changes on 2025-12-15

Reviewers

mminutoli

Assignees

No one assigned

Labels

None yet

Milestone

No milestone