tests/L0/run_optimizers/test_fused_optimizer.py · 81f8ba7952712c80e260cdb80e3646caf0a94811 · OpenDAS / apex

Masaki Kozuki authored Jun 22, 2022

* add temporary dispatch of double, float, half, bfloat16

* fusedadam of bfloat16

* Add bfloat16 path to FusedAdam

81f8ba79

test_fused_optimizer.py 11.2 KB