jax
aac9e9d2 - Add shared utility function get_rocm_version to test_util.py

Commit

75 days ago

Add shared utility function get_rocm_version to test_util.py

References

#556 - Porting v0.7.1 to v0.8.0

#579 - Create rocm-test-requirements.txt

#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

#591 - Adding run_pytest_rocm.sh

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

#670 - Test pytest abort plugin

#679 - Add ROCm-specific binary encoding for test_struct_encoding_determinism

Author

charleshofer

charleshofer

Committer

charleshofer

charleshofer

Parents

Loading