1. 10 May, 2023 1 commit
  2. 23 Mar, 2023 1 commit
    • ver217's avatar
      [lazyinit] combine lazy tensor with dtensor (#3204) · f8289d42
      ver217 authored
      * [lazyinit] lazy tensor add distribute
      
      * [lazyinit] refactor distribute
      
      * [lazyinit] add test dist lazy init
      
      * [lazyinit] add verbose info for dist lazy init
      
      * [lazyinit] fix rnn flatten weight op
      
      * [lazyinit] polish test
      
      * [lazyinit] polish test
      
      * [lazyinit] fix lazy tensor data setter
      
      * [lazyinit] polish test
      
      * [lazyinit] fix clean
      
      * [lazyinit] make materialize inplace
      
      * [lazyinit] refactor materialize
      
      * [lazyinit] refactor test distribute
      
      * [lazyinit] fix requires_grad
      
      * [lazyinit] fix tolist after materialization
      
      * [lazyinit] refactor distribute module
      
      * [lazyinit] polish docstr
      
      * [lazyinit] polish lazy init context
      
      * [lazyinit] temporarily skip test
      
      * [lazyinit] polish test
      
      * [lazyinit] add docstr
      f8289d42
  3. 17 Mar, 2023 1 commit
    • ver217's avatar
      [lazyinit] add correctness verification (#3147) · 6ae8ed04
      ver217 authored
      * [lazyinit] fix shared module
      
      * [tests] add lazy init test utils
      
      * [tests] add torchvision for lazy init
      
      * [lazyinit] fix pre op fn
      
      * [lazyinit] handle legacy constructor
      
      * [tests] refactor lazy init test models
      
      * [tests] refactor lazy init test utils
      
      * [lazyinit] fix ops don't support meta
      
      * [tests] lazy init test timm models
      
      * [lazyinit] fix set data
      
      * [lazyinit] handle apex layers
      
      * [tests] lazy init test transformers models
      
      * [tests] lazy init test torchaudio models
      
      * [lazyinit] fix import path
      
      * [tests] lazy init test torchrec models
      
      * [tests] update torch version in CI
      
      * [tests] revert torch version in CI
      
      * [tests] skip lazy init test
      6ae8ed04