ColossalAI/tests/test_fx
Frank Lee f7878f465c
[fx] supported model tracing for huggingface bert (#1201)
* [fx] supported model tracing for huggingface bert

* polish test
2022-07-05 13:19:57 +08:00
..
test_tracer [fx] supported model tracing for huggingface bert (#1201) 2022-07-05 13:19:57 +08:00
test_coloproxy.py [fx] added unit test for coloproxy (#1119) 2022-06-15 15:27:51 +08:00
test_parallel_1d.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00