jax
Backport memory space export test from JAX v0.9.0 to v0.8.2.
#700

Merged

Backport memory space export test from JAX v0.9.0 to v0.8.2. #700

magaonka-amd merged 1 commit into rocm-jaxlib-v0.8.2 from serial-compat-test-downstream-v0.8.2

tsrw2048 requested a review 246 days ago

tsrw2048 added open-upstream

tsrw2048 added cherry-pick-candidate

Modified memory space export test to run on ROCm (for some tests).

4629b349

tsrw2048 force pushed from c6879106 to 4629b349 246 days ago

magaonka-amd approved these changes on 2026-02-09

magaonka-amd merged 76568193 into rocm-jaxlib-v0.8.2 246 days ago

Reviewers

magaonka-amd

Assignees

No one assigned

Labels

open-upstream cherry-pick-candidate

Milestone

No milestone