ColossalAI/tests/test_auto_parallel
YuliangLiu0306 fea3cb661c
[autoparallel] support addmm in tracer and solver (#1961)
* [fx] patch addmm

* [autoparallel] support addmm in tracer and solver
2022-11-16 14:59:18 +08:00
..
test_tensor_shard [autoparallel] support addmm in tracer and solver (#1961) 2022-11-16 14:59:18 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00