Xuanlei Zhao 10c61de2f7 [autochunk] support vit (#3084) ...

support vit for autochunk
* support some new ops for vit
* fix some bugs
* add test for vit

2023-03-10 10:23:26 +08:00

..

test_autochunk_alphafold

[autochunk] refactor chunk memory estimation (#2762)

2023-03-08 16:22:30 +08:00

test_autochunk_diffuser

[autochunk] support vit (#3084)

2023-03-10 10:23:26 +08:00

test_autochunk_transformer

[autochunk] refactor chunk memory estimation (#2762)

2023-03-08 16:22:30 +08:00

test_autochunk_vit

[autochunk] support vit (#3084)

2023-03-10 10:23:26 +08:00