GitLab

* [autoparallel] refactor tracer to fix bias addition issue

* [fx] support module with bias addition

* create bias_addition_module

* refactor file structure

* polish code

* fix unit test

* [autoparallel] added matmul handler

* polish code

* [autoparallel] first move.

* [autoparallel] add solver rotor.

* [autoparallel] add ckpt solvers.

* [autoparallel] modify codegen.

* [fx] fix annotation in test.

* [fx] remove check.

* [autoparallel] polish docstring.

* [fx] refactor MetaTensor.

Co-authored-by: github-actions <github-actions@github.com>

* [autoparallel] update CommSpec to CommActions

* polish code

* [autoparallel] add numerical test for node strategies

* polish code

* polish code

* [fx] change memory.py to memory_utils.py.

* [fx] add shard utils.

* [fx] fix import.

* [fx] check code style.

* [fx] add comment.

* [autoparallel] first move.

* [fx] add time computations.

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>

* [autoparallel] refactor the runtime apply pass and add doc string to passes

* fix unit test

* polish

* [autoparallel] added binary elementwise node handler

* polish code

* [autoparallel] shard param and buffer as expected

* fix unit test issue

* [fx] test tracer on diffuser modules.

* [fx] shorter seq_len.

* Update requirements-test.txt

* [autoparallel] fixed wrong sharding strategy in conv handler

* polish code

* [autoparallel] fixed wrong generated strategy for dot op

* polish code

* [autoparallel] handled illegal sharding strategy in shape consistency

* polish code

* [autoparallel] handled illegal strategy in node handler

* polish code

* [fx/profiler] add test.

* [fx] fix file names.

* [fx] add docstring and comment.

* [fx] polish profiler.py.

* [fx] fix import errors.

* [fx] fix profiler.

* [fx] fix names.

* [autoparallel] handled illegal sharding strategy

* polish code