jax
b7a041cc - [Pallas:MGPU] Add a diagram outlining GPU memory spaces

Commit

273 days ago

[Pallas:MGPU] Add a diagram outlining GPU memory spaces

References

#392 - CI: 04/30/25 upstream sync

#518 - [0.6.0-UT] Adjust tolerance for reciprocal in unary primitives test

#519 - [0.6.0-UT] Fix Test Shard

#594 - Add CI workflow for Pytest ROCm

Author

apaszke

apaszke

Committer

charleshofer

charleshofer

Parents

Loading