ColossalAI/colossalai/_analyzer/_subclasses
Hongxin Liu dac127d0ee
[fx] fix meta tensor registration (#3589)
* [meta] fix torch 1.13.1

* [meta] fix torch 2.0.0

* [meta] fix torch 1.13.0

* [meta] polish code
2023-04-18 16:20:36 +08:00
..
__init__.py
_meta_registration.py
_monkey_patch.py
flop_tensor.py
meta_tensor.py