Fix/pallas tests shared memory#581

Open

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

fix/pallas-tests-shared-memory

Commits on Dec 29, 2025

- Fix `gpu_ops_test.py`: Implement block size reduction code. Dynamically reduce the block sizes until it fits the memory.
AratiGanesh
committed