14 Commits (8241c0c054b38a109ed3ce7be1052a1e600b8471)

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