GitLab

[autoparallel] Patch meta information of `torch.nn.functional.softmax` and `torch.nn.Softmax` (#2674)

* [autoparallel] softmax metainfo

* [autoparallel] softmax metainfo

* [workflow] added doc build test

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* [release] v0.2.3

* polish code

* [autoparallel] layernorm metainfo patch

* [autoparallel] polish test

* [tutorial] polish readme.md

* [example] Update README.md

* [doc] fix typo of BLOOM

* [autoparallel] refactor handlers which reshape input tensors

* polish

* [autoparallel] matmul metainfo

* [auto_parallel] remove unused print

* [tests] skip test_matmul_handler when torch version is lower than 1.12.0

* add alphafold benchmark

* renae alphafold test

* rename tests

* rename diffuser

* renme

* rename

* update transformer

* update benchmark

* update benchmark

* update bench memory

* update transformer benchmark

* rename

* support diffuser

* support unet metainfo prop

* fix bug and simplify code

* update linear and support some op

* optimize max region search, support conv

* update unet test

* support some op

* support groupnorm and interpolate

* update flow search

* add fix dim in node flow

* fix utils

* rename

* support diffusion

* update diffuser

* update chunk search

* optimize imports

* import

* finish autochunk

* [workflow] fixed the test coverage report

* polish code

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

* [release] v0.2.1

* polish code

* [workflow] added cuda extension build test before release

* polish code