jax
ac4ba1cb - Modified memory space export test to run on ROCm (for some tests). (#698)

Commit

103 days ago

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

References

#671 - Rocm jaxlib v0.9.0

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

#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)

Author

tsrw2048

tsrw2048

Parents

Loading