colossalai/tensor/d_tensor/utils.py · 7bc5a8e338a1240070ca82f96721c451060f1a70 · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[DTensor] implement layout converter (#3055) · 8e4e8601
YuliangLiu0306 authored Mar 10, 2023
```
* [DTensor] refactor LayoutConverter for DTensor

* polish code

* polish docstring
```
8e4e8601

utils.py 3.19 KB