DeepSpeed
compute global norm on device
#5125
Merged

Commits
  • compute global norm on device
    Liran Bachar committed 2 years ago
  • code review comments
    Liran Bachar committed 2 years ago
  • code review comments
    Liran Bachar committed 2 years ago
  • reuse graph code path for norm computation
    Liran Bachar committed 2 years ago
  • Fix formatting
    loadams committed 2 years ago
  • fix return value device to match input device
    Liran Bachar committed 2 years ago
  • Merge branch 'master' into global_norm_on_device
    loadams committed 2 years ago
Loading