Commits · cac061a10d953372e4dffd3b5b5b021cf70f58f4 · OpenDAS / apex

08 Mar, 2019 2 commits
- Code cleanup, add fused fp16 read / write · cac061a1
  Simon Layton authored Mar 08, 2019
```
Fuse in fp16 gradient -> fp32 convert
Additional option fp16 weight copy written out
```
  cac061a1
- Fused multi-tensor SGD · cadad920
  Simon Layton authored Mar 08, 2019
```
Initial implementation, all fp32
Tested against torch.optim.sgd
```
  cadad920