github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/_analyzer/fx/tracer/__init__.py

3 lines
63 B
Python
Raw Normal View History

[FX] refactor experimental tracer and adapt it with hf models (#3157) * pass gpt trace and meta_prop * pass t5 trace and meta_prop * [FX] refactor experimental tracer and adapt it with hf models * pass all mainstream model zoo * fix CI * fix CI * fix CI * fix CI * fix CI * fix CI * fix CI * fix CI * skip tests * fix CI * using packaging version * polish
2023-03-22 02:40:33 +00:00
from .bias_addition import *
from .custom_leaf_module import *