ColossalAI/tests/test_utils/test_lazy_init
ver217 f8289d4221
[lazyinit] combine lazy tensor with dtensor (#3204)
* [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
2023-03-23 10:53:06 +08:00
..
test_distribute.py [lazyinit] combine lazy tensor with dtensor (#3204) 2023-03-23 10:53:06 +08:00
test_models.py [lazyinit] add correctness verification (#3147) 2023-03-17 13:49:04 +08:00
utils.py [lazyinit] combine lazy tensor with dtensor (#3204) 2023-03-23 10:53:06 +08:00