digger yu
|
e2d81eba0d
|
[nfc] fix typo colossalai/ applications/ (#3831)
* fix typo colossalai/autochunk auto_parallel amp
* fix typo colossalai/auto_parallel nn utils etc.
* fix typo colossalai/auto_parallel autochunk fx/passes etc.
* fix typo docs/
* change placememt_policy to placement_policy in docs/ and examples/
* fix typo colossalai/ applications/
|
2 years ago |
digger yu
|
9265f2d4d7
|
[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)
* fix typo colossalai/autochunk auto_parallel amp
* fix typo colossalai/auto_parallel nn utils etc.
|
2 years ago |
YuliangLiu0306
|
0f392d7403
|
[autoparallel] find repeat blocks (#2854)
* [autoparallel] find repeat blocks
* polish
* polish
* polish
|
2 years ago |
binmakeswell
|
a881d6d000
|
Revert "[NFC] polish code format" (#2372)
|
2 years ago |
LuGY
|
e11a005c02
|
[NFC] polish colossalai/auto_parallel/tensor_shard/utils/factory.py code style (#2349)
|
2 years ago |
YuliangLiu0306
|
1c1fe44305
|
[autoparallel] adapt solver with self attention (#2037)
* [autoparallel] adapt solver with self attention
* polish code
|
2 years ago |
YuliangLiu0306
|
1438993113
|
[autoparallel] add experimental view handler (#2011)
* [autoparallel] add experimental view handler
* polish
* polish
* polish code
* rename variables
|
2 years ago |
YuliangLiu0306
|
155891113e
|
[autoparallel] use pytree map style to process data (#1989)
|
2 years ago |
YuliangLiu0306
|
e34e850a4c
|
[autoparallel]add essential CommActions for broadcast oprands (#1793)
|
2 years ago |
Frank Lee
|
f3f19a5c47
|
[autoparallel] added matmul handler (#1763)
* [autoparallel] added matmul handler
* polish code
|
2 years ago |
Frank Lee
|
f9a613d660
|
[autoparallel] added binary elementwise node handler (#1758)
* [autoparallel] added binary elementwise node handler
* polish code
|
2 years ago |
Frank Lee
|
262652c8bc
|
[autoparallel] added addbmm handler (#1751)
|
2 years ago |
Frank Lee
|
474111ecb5
|
[autoparallel] fixed wrong sharding strategy in conv handler (#1747)
* [autoparallel] fixed wrong sharding strategy in conv handler
* polish code
|
2 years ago |
Frank Lee
|
8b8937d901
|
[autoparallel] fixed wrong generated strategy for dot op (#1746)
* [autoparallel] fixed wrong generated strategy for dot op
* polish code
|
2 years ago |
Frank Lee
|
88a79814fb
|
[autoparallel] handled illegal strategy in node handler (#1743)
* [autoparallel] handled illegal strategy in node handler
* polish code
|
2 years ago |
Frank Lee
|
eee84908d4
|
[autoparallel] handled illegal sharding strategy (#1728)
* [autoparallel] handled illegal sharding strategy
* polish code
|
2 years ago |
Frank Lee
|
6c331a5a09
|
[autoparallel] refactored the autoparallel module for organization (#1706)
* [autoparallel] refactored the autoparallel module for organization
* polish code
|
2 years ago |