jax
1f3139e0 - Merge pull request #34601 from ROCm:phambinh/ci_rocm_nn_tests

Commit

2 days ago

Merge pull request #34601 from ROCm:phambinh/ci_rocm_nn_tests PiperOrigin-RevId: 875282366

References

#26604 - [ROCm][pallas:triton] Fix atomic min/max lowering for uint and float types

#34601 - [ROCm] Enable neural network tests on ROCm

#34702 - [ROCm] Add script for running bazel tests on ROCm

#34722 - [ROCm] ToT ROCm Unit Test Skips

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading