jax
e540fe35 - Add devices fingerprint into the base launch ID. This ensures that the launch ID is unique if the same executable is executed on different subset of devices.

Commit

154 days ago

Add devices fingerprint into the base launch ID. This ensures that the launch ID is unique if the same executable is executed on different subset of devices. PiperOrigin-RevId: 789826923

References

#30692 - Add devices fingerprint into the base launch ID. This ensures that the launch ID is unique if the same executable is executed on different subset of devices.

#31381 - Remove old ROCm build code

#31720 - Fix ann_test.py numerical bug in target reshape

#31768 - [ROCm] Support lowering through PJRT_Triton_Extension

#32115 - Relax version requirements for ROCm Jax Plugin wheels

#33157 - Resolve undefined behavior in bitshift unit test

#33186 - Make nvidia version data optional for ROCm builds

#579 - Create rocm-test-requirements.txt

#580 - GESVDJ support for ROCm GPUs in JAX

#581 - Fix/pallas tests shared memory

#584 - Use plain bazel to test jax, use hermetic rocm dependency

#585 - update a test for checking zero ROCm GPU event

#34135 - [ROCm] update to test if there are GPU events when doing profiling on…

Author

Google-ML-Automation

Committer

Google-ML-Automation

Parents

6330af95

jax e540fe35 - Add devices fingerprint into the base launch ID. This ensures that the launch ID is unique if the same executable is executed on different subset of devices.

jax
e540fe35 - Add devices fingerprint into the base launch ID. This ensures that the launch ID is unique if the same executable is executed on different subset of devices.