csrc/multi_tensor_adam.cu · 96850dfa73661f61b37d8b984efca17876f7fe77 · 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

multi_tensor_adam.cu 4.42 KB