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 [analyzer] a minimal implementation of static graph analyzer (#2852) 2023-03-10 13:21:05 +08:00
_meta_registration.py [fx] fix meta tensor registration (#3589) 2023-04-18 16:20:36 +08:00
_monkey_patch.py [gemini] gemini supports lazy init (#3379) 2023-04-12 16:03:25 +08:00
flop_tensor.py [autoparallel]integrate auto parallel feature with new tracer (#3408) 2023-04-04 17:40:45 +08:00
meta_tensor.py [analyzer] a minimal implementation of static graph analyzer (#2852) 2023-03-10 13:21:05 +08:00