jax
f6fd826f - Enable testDotAlgorithm BF16/TF32 tests on ROCm (#628)

Commit

3 days ago

Enable testDotAlgorithm BF16/TF32 tests on ROCm (#628)

References

#628 - Enable testDotAlgorithm BF16/TF32 tests on ROCm

#654 - Skip test_batch_axis_sharding_jvp on ROCm

Author

magaonka-amd

magaonka-amd

Parents

Loading