csrc/multi_tensor_sgd_kernel.cu · 274cc0630286d598e14860cfda91752d0840f712 · OpenDAS / apex

set device guard for multi tensor optimizer implementations (#927) · 274cc063

ngimel authored Aug 05, 2020

* add device guards to the optimizers

* add untracked file

* set deviceGuard in multi_tensor_apply

* address review comments; fix lamb

* indent

* typo

274cc063

multi_tensor_sgd_kernel.cu 7.93 KB

Replace multi_tensor_sgd_kernel.cu