Commits · 57bfa7caba01bf96befd292e02ffe79fc744f1bd · OpenDAS / Megatron-LM

30 Sep, 2022 1 commit
- Perform distributed optimizer's all-gather in param dtype (instead of grad dtype) · 57bfa7ca
  Lawrence McAfee authored Sep 30, 2022
  
  57bfa7ca
02 Sep, 2022 1 commit
- Memory safety checks were incorrect for the tokens_to_generate=0 case · e38d41ca
  rprenger authored Sep 02, 2022
  
  e38d41ca
16 Aug, 2022 2 commits
- fixed grad scalar warning for bf16 · aaa5715a
  Mohammad Shoeybi authored Aug 16, 2022
  
  aaa5715a
- fixed grad scalar warning so it only prints it for fp16 · 27bc1338
  mshoeybi authored Aug 15, 2022
  
  27bc1338
10 Aug, 2022 1 commit
- Timing levels · 77efcccb
  Mohammad Shoeybi authored Aug 10, 2022
  
  77efcccb
06 Aug, 2022 1 commit
- fix a bug for size mismatch · b7b2d6a9
  Peng Xu authored Aug 05, 2022
  
  b7b2d6a9
28 Jul, 2022 1 commit
- support for all mask in fused kernel + avoiding inplace operation in bwd pass · 76db9583
  Vijay Korthikanti authored Jul 28, 2022
  
  76db9583
26 Jul, 2022 1 commit
- added a flag to be able to switch between pytorch and ring exchange p2p · e36cdd79
  Mohammad Shoeybi authored Jul 26, 2022
  
  e36cdd79
21 Jul, 2022 1 commit
- Remove deprecated destination argument to state_dict functions and make all... · 928a200c
  Jared Casper authored Jul 21, 2022
```
Remove deprecated destination argument to state_dict functions and make all arguments keyword to avoid warnings.
```
  928a200c
18 Jul, 2022 1 commit
- Adding checks for total number of tokens to keep server from crashing · cfe2c2be
  rprenger authored Jul 18, 2022
  
  cfe2c2be
15 Jul, 2022 2 commits
- remove debug lines for printing · 2e65bee5
  peng xu authored Jul 15, 2022
  
  2e65bee5
- fix pipeline parallel for beam search · e5034150
  peng xu authored Jul 15, 2022
  
  e5034150
24 Jun, 2022 2 commits
- Fixing up checkpointing. · 2725dc0b
  Jared Casper authored Jun 24, 2022
  
  2725dc0b
- Remove unused function. · 4eb802c4
  Jared Casper authored Jun 24, 2022
  
  4eb802c4
21 Jun, 2022 1 commit
- Changing name of parameters from factual_decay to top_p_decay · ed6806ac
  rprenger authored Jun 21, 2022
  
  ed6806ac
08 Jun, 2022 6 commits
- Adding top_p decay and bound for factual sampling from Factuality Enhanced LMs · b49349ec
  rprenger authored Jun 08, 2022
  
  b49349ec
- fixed load_checkpoint(). · 74ba3797
  Lawrence McAfee authored Jun 08, 2022
  
  74ba3797
- error fixes & tested. · b178e6fc
  Lawrence McAfee authored Jun 08, 2022
  
  b178e6fc
- added backwards compatibility to checkpointing.py. · 977efdfb
  Lawrence McAfee authored Jun 08, 2022
  
  977efdfb
- re-added grad dtype assertion in clip_grads. · ef63acd2
  Lawrence McAfee authored Jun 08, 2022
  
  ef63acd2
- removed count-zeros debuggables. · 9044bc42
  Lawrence McAfee authored Jun 08, 2022
  
  9044bc42
01 Jun, 2022 1 commit
- testing fix w/ clip_grads. · 7fccd6a1
  Lawrence McAfee authored Jun 01, 2022
  
  7fccd6a1
26 May, 2022 1 commit
- bug fix for get_data_parallel_src_rank · 9ad1944d
  Vijay Korthikanti authored May 26, 2022
  
  9ad1944d
25 May, 2022 1 commit
- separate file for license · da11c982
  Peng Xu authored May 25, 2022
  
  da11c982
24 May, 2022 7 commits
- resolved review comments · 739cb43d
  Vijay Korthikanti authored May 24, 2022
  
  739cb43d
- allreduce layernorm fixes. · de593298
  Lawrence McAfee authored May 24, 2022
  
  de593298
- fixed for new jit warmup changes. · cf1c7848
  Lawrence McAfee authored May 24, 2022
  
  cf1c7848
- preallocating global buffer to avoid memory fragmentation · 9dc3c42a
  Vijay Korthikanti authored May 24, 2022
  
  9dc3c42a
- added timer for layernorm all-reduce. · aff99e32
  Lawrence McAfee authored May 24, 2022
  
  aff99e32
- moved SP's layernorm all-reduce into optimizer.py. · ee57e086
  Lawrence McAfee authored May 24, 2022
  
  ee57e086
- fix for sequence parallelism in bert pooling · 8474e6e5
  Vijay Korthikanti authored May 24, 2022
  
  8474e6e5
20 May, 2022 2 commits
- avoiding sequence parallelism on the pooler · 3f91f09b
  Vijay Korthikanti authored May 20, 2022
  
  3f91f09b
- bert regression fixes · 6aaafee6
  Vijay Korthikanti authored May 20, 2022
  
  6aaafee6
18 May, 2022 1 commit
- t5 regression fixes · 5d647381
  Vijay Korthikanti authored May 18, 2022
  
  5d647381
17 May, 2022 4 commits
- address review comment · a7a12f82
  Vijay Korthikanti authored May 17, 2022
  
  a7a12f82
- address review comments · d19e8ad7
  Vijay Korthikanti authored May 16, 2022
  
  d19e8ad7
- address review comments · bbab79f8
  Vijay Korthikanti authored May 16, 2022
  
  bbab79f8
- address review comments · 356eb36a
  Vijay Korthikanti authored May 16, 2022
  
  356eb36a
16 May, 2022 2 commits
- address review comment · 214a9a08
  Vijay Korthikanti authored May 16, 2022
  
  214a9a08
- cleanup imports in transformer.py · 28b6ffce
  Vijay Korthikanti authored May 16, 2022
  
  28b6ffce