jax
a2b972f2 - register rocm platform to mx datatype lowering path (#567)

Commit

101 days ago

register rocm platform to mx datatype lowering path (#567)

References

#567 - register rocm platform to mx datatype lowering path

#579 - Create rocm-test-requirements.txt

#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

#731 - update test

Author

ScXfjiang

ScXfjiang

Parents

Loading