DeepSpeed
7be128af - Allocating CPU memory directly on CPU without transfering them from GPU (#360)

Comment changes are shownComment changes are hidden
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
Author
Parents
  • deepspeed
    • ops/adam
      • File
        cpu_adam.py
    • runtime/zero
      • File
        stage1.py
      • File
        stage2.py
  • tests/unit
    • File
      test_adam_acuracy.py