jax
552778a4 - Add rocm test requirements file (#570)

Commit

64 days ago

Add rocm test requirements file (#570)

References

#570 - Add rocm test requirements file

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

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

#591 - Adding run_pytest_rocm.sh

#599 - Use xla from rocm-jaxlib-v0.8.0 release branch

#670 - Test pytest abort plugin

#701 - Skip test_tridiagonal_solve_grad test 0.8.0

Author

AratiGanesh

AratiGanesh

Parents

Loading