github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-21 09:29:47 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
feature/stable-diffusion
ColossalAI/colossalai/_analyzer/_subclasses
History
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