Commit Graph

14 Commits (50104ab340e6c7067fbaaf9b47c608eb828aa95b)

Author SHA1 Message Date
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files (#4752)
1 year ago
CZYCW 4cadb25b96 [NFC] policy colossalai/fx/proxy.py code style (#3269)
2 years ago
Super Daniel ea1a95b8b9
[hotfix] fix coloproxy typos. (#1519)
2 years ago
ver217 d068af81a3
[doc] update rst and docstring (#1351)
2 years ago
YuliangLiu0306 942c8cd1fb
[fx] refactor tracer to trace complete graph (#1342)
2 years ago
YuliangLiu0306 4631fef8a0
[fx]refactor tracer (#1335)
2 years ago
Frank Lee fb35460595
[fx] added ndim property to proxy (#1253)
2 years ago
Frank Lee 5da87ce35d
[fx] added testing for all albert variants (#1211)
2 years ago
Frank Lee 2d13a45a3b
[fx] added testing for all gpt variants (#1210)
2 years ago
Frank Lee 426a279ce7
[fx] added testing for all bert variants (#1207)
2 years ago
Frank Lee f7878f465c
[fx] supported model tracing for huggingface bert (#1201)
2 years ago
Frank Lee 6d86f1bc91
[fx] supported data-dependent control flow in model tracing (#1185)
2 years ago
Frank Lee 16302a5359
[fx] added unit test for coloproxy (#1119)
2 years ago
Frank Lee e1620ddac2
[fx] added coloproxy (#1115)
2 years ago