DeepSpeed
7be128af
- Allocating CPU memory directly on CPU without transfering them from GPU (#360)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
4 years ago
Allocating CPU memory directly on CPU without transfering them from GPU (#360) * Allocating CPU memory directly on CPU without transfering them from GPU * formatting fixes
References
#360 - Allocating CPU memory directly on CPU without transfering them from GPU
Author
samyam
Parents
077cfd47
Files
4
deepspeed
ops/adam
cpu_adam.py
runtime/zero
stage1.py
stage2.py
tests/unit
test_adam_acuracy.py
Loading