[lazyinit] refactor lazy tensor and lazy init ctx (#3131)

* [lazyinit] refactor lazy tensor and lazy init ctx * [lazyinit] polish docstr * [lazyinit] polish docstr

[lazyinit] refactor lazy tensor and lazy init ctx (#3131)
* [lazyinit] refactor lazy tensor and lazy init ctx * [lazyinit] polish docstr * [lazyinit] polish docstr
ed8f60b9 · ver217 · GitHub · 86ac782d · ed8f60b9
Unverified Commit ed8f60b9 authored Mar 14, 2023 by ver217 Committed by GitHub Mar 14, 2023
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 170 additions and 214 deletions

colossalai/utils/model/experimental.py colossalai/utils/model/experimental.py +170 -214

No files found.
--- a/colossalai/utils/model/experimental.py
+++ b/colossalai/utils/model/experimental.py