GitLab

* [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

Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>

* add chunk manager init function

* fix unit tests

* add comment

* add flush=True

* update pre-commit

* update pre-commit