ColossalAI/tests/test_fx
Frank Lee 426a279ce7
[fx] added testing for all bert variants (#1207)
* [fx] added testing for all bert variants

* polish code
2022-07-06 10:50:49 +08:00
..
test_tracer [fx] added testing for all bert variants (#1207) 2022-07-06 10:50:49 +08:00
test_coloproxy.py
test_parallel_1d.py