colossalai/utils/model/colo_init_context.py · 32291dd73f92843b4f221226a95f2f7048b1766b · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[Tensor] add module handler for linear (#1021) · 32291dd7
Ziyue Jiang authored May 26, 2022
```
* add module spec for linear

* polish

* polish

* polish
```
32291dd7

colo_init_context.py 7.45 KB