Commits · 34f55429186b33bbf5f3d588f3251d388f44fa56 · OpenDAS / Megatron-LM

26 Jan, 2022 1 commit
- further clarified viewless tensor comment in transformer.py · d16e2a24
  Lawrence McAfee authored Jan 26, 2022
  
  d16e2a24
25 Jan, 2022 1 commit
- limit 'make_viewless_tensor()' to case of micro_batch_size == 1; added comment · 24369dd6
  Lawrence McAfee authored Jan 25, 2022
  
  24369dd6
20 Jan, 2022 1 commit
- address review comments · 798b6a64
  Vijay Korthikanti authored Jan 20, 2022
  
  798b6a64
19 Jan, 2022 2 commits
- using all_gather instead of gather (nccl does not support gather) · b037a69e
  Vijay Korthikanti authored Jan 19, 2022
  
  b037a69e
- flag for data parallel random initialization · a7ee77ea
  Vijay Korthikanti authored Jan 19, 2022
  
  a7ee77ea
15 Jan, 2022 2 commits
- minor fix · fd8dd9c0
  Vijay Korthikanti authored Jan 14, 2022
  
  fd8dd9c0
- checkpointing rng_state of all data parallel ranks · 09d38884
  Vijay Korthikanti authored Jan 14, 2022
  
  09d38884
14 Jan, 2022 1 commit
- address review comments · 343dc97a
  Vijay Korthikanti authored Jan 14, 2022
  
  343dc97a
13 Jan, 2022 7 commits
- handle output_tensor == None · 7f1c5917
  Lawrence McAfee authored Jan 13, 2022
  
  7f1c5917
- more cleanup · 8fc5e323
  Lawrence McAfee authored Jan 13, 2022
  
  8fc5e323
- renamed 'free' -> 'deallocate' · c1e4526b
  Lawrence McAfee authored Jan 13, 2022
  
  c1e4526b
- removed uses are args.deallocate_pipeline_output · d10f81c5
  Lawrence McAfee authored Jan 13, 2022
  
  d10f81c5
- removed argument flag; also, converted torch.FloatTenstor -> torch.empty in previous commit · 18846a0a
  Lawrence McAfee authored Jan 13, 2022
  
  18846a0a
- fixed bug for no-interleave pipeline schedule · 1f387c2c
  Lawrence McAfee authored Jan 13, 2022
  
  1f387c2c
- minor fixes · 3b29f0e7
  Vijay Korthikanti authored Jan 13, 2022
  
  3b29f0e7
12 Jan, 2022 1 commit
- Phase1 merge: vit optimizations + dataset enhancements + scaled_softmax kernel · 7a77abd9
  Vijay Korthikanti authored Jan 12, 2022
  
  7a77abd9
11 Jan, 2022 4 commits
- removed several 'assert_viewless_tensor' calls. · 0c8e8cce
  Lawrence McAfee authored Jan 11, 2022
  
  0c8e8cce
- added comments · a1fe4805
  Lawrence McAfee authored Jan 11, 2022
  
  a1fe4805
- partially cleaned · 806422e5
  Lawrence McAfee authored Jan 11, 2022
  
  806422e5
- jan 11 alpha · 05042081
  Lawrence McAfee authored Jan 11, 2022
  
  05042081
10 Jan, 2022 1 commit
- loss matches; memory savings for multi-node (tested n3, n16) · 270d6412
  Lawrence McAfee authored Jan 10, 2022
  
  270d6412
08 Jan, 2022 2 commits
- jan 7 beta · b6d4369b
  Lawrence McAfee authored Jan 07, 2022
  
  b6d4369b
- more iterating on 'viewless tensor' methods · ed0c8714
  Lawrence McAfee authored Jan 07, 2022
  
  ed0c8714
07 Jan, 2022 1 commit
- debugging make_standalone_tensor(), safely_set_tensor_data_attr() · 5422d23a
  Lawrence McAfee authored Jan 07, 2022
  
  5422d23a
04 Jan, 2022 2 commits
- added flag to control deallocation of pipeline outputs · de1ae5b2
  Lawrence McAfee authored Jan 04, 2022
  
  de1ae5b2
- update comment · d439b417
  Vijay Korthikanti authored Jan 04, 2022
  
  d439b417
31 Dec, 2021 1 commit
- simplify code · 9da6e975
  Vijay Korthikanti authored Dec 30, 2021
  
  9da6e975
30 Dec, 2021 2 commits
- avoid allocation of word embedding for head in T5 pp=2 case · ade99d61
  Vijay Korthikanti authored Dec 30, 2021
  
  ade99d61
- fixing the corner case pp=2 · 26ea8314
  Vijay Korthikanti authored Dec 30, 2021
  
  26ea8314
21 Dec, 2021 1 commit
- comments · 81ad628e
  Vijay Korthikanti authored Dec 21, 2021
  
  81ad628e
17 Dec, 2021 2 commits
- minor fixes · f2bf5a56
  Vijay Korthikanti authored Dec 17, 2021
  
  f2bf5a56
- pipeline_fixes · 17843605
  Vijay Korthikanti authored Dec 17, 2021
  
  17843605
10 Dec, 2021 1 commit
- output tensor now allocated directly on gpu. · 85876092
  Lawrence McAfee authored Dec 10, 2021
  
  85876092
08 Dec, 2021 1 commit
- added docstrings to new methods · e33d63cc
  Lawrence McAfee authored Dec 08, 2021
  
  e33d63cc
07 Dec, 2021 2 commits
- working: interleaving; free_output_tensor() now handles none/tensor/list · 2f25c570
  Lawrence McAfee authored Dec 07, 2021
  
  2f25c570
- working for pure pipeline parallelism, w/ no interleaving · 86da10e9
  Lawrence McAfee authored Dec 07, 2021
  
  86da10e9
02 Dec, 2021 2 commits
- update training.py · e1b3fc8c
  zihanl authored Dec 01, 2021
  
  e1b3fc8c
- update tokenizer.py · 67fabc55
  zihanl authored Dec 01, 2021
  
  67fabc55
26 Nov, 2021 1 commit
- fix training.py · b1a6d73b
  zihanl authored Nov 25, 2021
  
  b1a6d73b
24 Nov, 2021 1 commit
- Add support for signal-based dynamic checkpointing · dbe6c728
  Szymon Migacz authored Nov 23, 2021
  
  dbe6c728