Commits · bf64c85c57f0e33e9907707117e0c581ea456e41 · OpenDAS / Megatron-LM

15 Feb, 2022 6 commits
- added 'all' world shards to gbuf map · bf64c85c
  Lawrence McAfee authored Feb 15, 2022
  
  bf64c85c
- updated zero_grad · eaa0c1df
  Lawrence McAfee authored Feb 15, 2022
  
  eaa0c1df
- allocated main params/grads · 33626179
  Lawrence McAfee authored Feb 15, 2022
  
  33626179
- created optimizer group shards · 525a8351
  Lawrence McAfee authored Feb 15, 2022
  
  525a8351
- modularized shard indexing · f498a6fe
  Lawrence McAfee authored Feb 15, 2022
  
  f498a6fe
- wip; switching to grad-buffer-centric design · cb6f96b6
  Lawrence McAfee authored Feb 15, 2022
  
  cb6f96b6
14 Feb, 2022 8 commits
- todo; align shards with model's contiguous buffer · a3f3c3ad
  Lawrence McAfee authored Feb 14, 2022
  
  a3f3c3ad
- copying model grad slices to main grad · 3f0bc681
  Lawrence McAfee authored Feb 14, 2022
  
  3f0bc681
- fix zero_grad; set_to_none = False · 6875dff5
  Lawrence McAfee authored Feb 14, 2022
  
  6875dff5
- tweaked slice index naming convention · 1215c420
  Lawrence McAfee authored Feb 14, 2022
  
  1215c420
- map param to originating virtual model; eventually move this to constructor · c5f93269
  Lawrence McAfee authored Feb 14, 2022
  
  c5f93269
- included original param index in map · 3ded2425
  Lawrence McAfee authored Feb 14, 2022
  
  3ded2425
- 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 5 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