jax
f893175a - Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

Commit

88 days ago

Fix v_pages quantization and adjust test params for ROCm compatibilit… (#560)

References

#560 - Fix v_pages quantization and adjust test params for ROCm compatibilit…

#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

#701 - Skip test_tridiagonal_solve_grad test 0.8.0

#708 - Skip test_batch_axis_sharding_jvp13 test 0.8.0

Author

phambinhfin

phambinhfin

Parents

Loading