jax
9e7b005e - Add libtpu date guard for failing test.

Commit

143 days ago

Add libtpu date guard for failing test. https://source.cloud.google.com/results/invocations/d6f7cee9-3f2a-4c40-bc1f-cd7463a797f5 PiperOrigin-RevId: 857271332

References

#664 - update with ROCm profiling tests with kernel_details

#671 - Rocm jaxlib v0.9.0

#699 - Skip testEighTinyNorm test 0.9.0

#725 - Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling

#732 - update test

#734 - Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU …

#739 - Fix: Scaled Matmul at rocm 9.0

#765 - jaxlib/gpu/rnn_kernels.cc: remove premature input_tensor_desc destruction (ROCM-21859)

#785 - Cherry-pick ROCm fixes from PR #740 to rocm-jaxlib-v0.9.0

Author

mwhittaker

Committer

mwhittaker

Parents

18b34a12

jax 9e7b005e - Add libtpu date guard for failing test.

jax
9e7b005e - Add libtpu date guard for failing test.