1. 15 Aug, 2023 3 commits
    • Jianghai's avatar
      [pipeline] Bert pipeline for shardformer and its tests (#4197) · 1094e0f0
      Jianghai authored
      * add pipeline forward
      
      * complete pipeline forward check
      
      * fix bert forward without pipeline
      
      * fix comments
      
      * discard useless line
      
      * add todo
      
      * clean prints
      
      * fix distribute layers
      1094e0f0
    • Hongxin Liu's avatar
      [shardformer] support lazy init (#4202) · 890774b2
      Hongxin Liu authored
      * [shardformer] support lazy init
      
      * [shardformer] linear support lazy init
      
      * [shardformer] embedding support lazy init
      
      * [shardformer] norm support lazy init
      
      * [shardformer] fused linear support lazy init
      
      * [test] update shardformer test layer
      
      * [test] shardformer with lazy init fit ddp
      
      * [lazy] hotfix deepcopy of param
      
      * [shardformer] fix bert policy and update test
      
      * [shardformer] fix bloom policy and update test
      
      * [shardformer] fix opt policy and update test
      
      * [shardformer] fix t5 policy and update test
      
      * [shardformer] fix gpt2 policy and update test
      
      * [shardformer] fix llama policy and update test
      890774b2
    • ver217's avatar
      [test] update shardformer tests · 2d6cc07f
      ver217 authored
      2d6cc07f
  2. 04 Jul, 2023 7 commits