tests/L0/run_optimizers/test_fused_optimizer.py · a651e2c24ecf97cbf367fd3f330df36760e1c597 · 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

test_fused_optimizer.py 10.2 KB

Replace test_fused_optimizer.py