Commits · a74e245cfca7b6a21dce0807eedede5318b0456e · OpenDAS / Megatron-LM

14 Feb, 2022 2 commits
- built local shard param index map · a74e245c
  Lawrence McAfee authored Feb 14, 2022
  
  a74e245c
- collect param offsets for contiguous grad buffer · f7232502
  Lawrence McAfee authored Feb 14, 2022
  
  f7232502
11 Feb, 2022 2 commits
- bit more progress · 5706ba42
  Lawrence McAfee authored Feb 11, 2022
  
  5706ba42
- studied float16 optimizer; more updates · f48e1f29
  Lawrence McAfee authored Feb 11, 2022
  
  f48e1f29
10 Feb, 2022 2 commits
- more work on Float16DistributedOptimizer · 49cca4d9
  Lawrence McAfee authored Feb 10, 2022
  
  49cca4d9
- working on Float16DistributedOptimizer · 329fe582
  Lawrence McAfee authored Feb 10, 2022
  
  329fe582
09 Feb, 2022 1 commit
- feb 9 alpha · 7dc8c475
  Lawrence McAfee authored Feb 09, 2022
  
  7dc8c475
29 Jan, 2022 2 commits
- typo fix · 2b628f96
  Vijay Korthikanti authored Jan 28, 2022
  
  2b628f96
- Revert incorrect fix. · cd499559
  Jared Casper authored Jan 28, 2022
  
  cd499559
28 Jan, 2022 2 commits
- more naming cleanup · 641408f5
  Vijay Korthikanti authored Jan 28, 2022
  
  641408f5
- changing class name AnnealingLR to OptimizerParamScheduler · 04ecc834
  Vijay Korthikanti authored Jan 28, 2022
  
  04ecc834
27 Jan, 2022 1 commit
- address review comments · 53931b8b
  Vijay Korthikanti authored Jan 27, 2022
  
  53931b8b
26 Jan, 2022 2 commits
- further clarified viewless tensor comment in transformer.py · d16e2a24
  Lawrence McAfee authored Jan 26, 2022
  
  d16e2a24
- address review comments · 8acbbe25
  Vijay Korthikanti authored Jan 26, 2022
  
  8acbbe25
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
24 Jan, 2022 1 commit
- minor fixes · 7e810e41
  Vijay Korthikanti authored Jan 24, 2022
  
  7e810e41
22 Jan, 2022 1 commit
- second phase of vision code merge · d8c85650
  Vijay Korthikanti authored Jan 21, 2022
  
  d8c85650
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 1 commit
- added flag to control deallocation of pipeline outputs · de1ae5b2
  Lawrence McAfee authored Jan 04, 2022
  
  de1ae5b2