github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai
History
YuliangLiu0306 2c4c7b3618
[autoparallel] add getattr handler (#1767) 
* [autoparallel] add getattr haandler

* polish code

* add extra processes for Parameters

* add unit test for param resharding cost

* add docstring and polish test
 		2022-11-03 12:31:33 +08:00
..
amp [doc] update rst and docstring (#1351) 2022-07-21 15:54:53 +08:00
auto_parallel [autoparallel] add getattr handler (#1767) 2022-11-03 12:31:33 +08:00
builder [NFC] polish colossalai/builder/__init__.py code style (#1560) 2022-09-08 22:11:04 +08:00
cli [hotfix] fix some bugs caused by size mismatch. (#1011) 2022-05-23 14:02:28 +08:00
communication [communication] add p2p_v2.py to support communication with List[Any] (#1407) 2022-08-09 11:40:04 +08:00
context updated tp layers 2022-11-02 12:19:38 +08:00
device [autoparallel] add numerical test for node strategies (#1760) 2022-10-27 10:42:54 +08:00
engine [engin/schedule] use p2p_v2 to recontruct pipeline_schedule (#1408) 2022-08-12 11:33:26 +08:00
fx [autoparallel] add getattr handler (#1767) 2022-11-03 12:31:33 +08:00
gemini [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786) 2022-11-02 16:11:34 +08:00
kernel [feat] add flash attention (#1762) 2022-10-26 16:15:52 +08:00
logging [doc] improved docstring in the logging module (#861) 2022-04-25 13:42:00 +08:00
nn [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786) 2022-11-02 16:11:34 +08:00
pipeline [Pipeline]Adapt to Pipelinable OPT (#1782) 2022-11-01 14:18:50 +08:00
registry Remove duplication registry (#1078) 2022-06-08 07:47:24 +08:00
tensor [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786) 2022-11-02 16:11:34 +08:00
testing [unittest] added doc for the pytest wrapper (#1704) 2022-10-14 10:56:17 +08:00
trainer [NFC] polish _checkpoint_hook.py code style (#1722) 2022-10-19 12:20:51 +08:00
utils [zero] add constant placement policy (#1705) 2022-10-14 17:53:16 +08:00
zero [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786) 2022-11-02 16:11:34 +08:00
__init__.py upgrade version to 0.1.11rc1 (#1739) 2022-10-19 11:26:00 +08:00
constants.py updated tp layers 2022-11-02 12:19:38 +08:00
core.py [Tensor] distributed view supports inter-process hybrid parallel (#1169) 2022-06-27 09:45:26 +08:00
global_variables.py updated tp layers 2022-11-02 12:19:38 +08:00
initialize.py [hotfix] remove potiential circle import (#1307) 2022-07-14 13:44:26 +08:00