github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai
History
ver217 ab8c6b4a0e
[zero] refactor memstats collector (#706) 
* refactor memstats collector

* fix disposable

* polish code
 		2022-04-11 10:46:08 +08:00
..
amp fix format (#570) 2022-04-06 11:40:59 +08:00
builder [NFC] polish colossalai/builder/builder.py code style (#662) 2022-04-06 11:40:59 +08:00
communication [NFC] polish colossalai/communication/utils.py code style (#656) 2022-04-06 11:40:59 +08:00
context [NFC] polish colossalai/context/process_group_initializer/initializer_sequence.py colossalai/context/process_group_initializer initializer_tensor.py code style (#639) 2022-04-06 11:40:59 +08:00
engine [zero] adapt zero hooks for unsharded module (#699) 2022-04-08 20:23:26 +08:00
kernel [NFC] polish colossalai/kernel/cuda_native/csrc/multi_tensor_adam.cu code style (#667) 2022-04-06 11:40:59 +08:00
logging Refactored docstring to google style 2022-03-29 17:17:47 +08:00
nn []Corrected 3d vocab parallel embedding (#707) 2022-04-11 10:17:55 +08:00
registry Refactored docstring to google style 2022-03-29 17:17:47 +08:00
testing [test] fixed rerun_on_exception and adapted test cases (#487) 2022-03-25 17:25:12 +08:00
trainer [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00
utils [zero] refactor memstats collector (#706) 2022-04-11 10:46:08 +08:00
zero [zero] refactor memstats collector (#706) 2022-04-11 10:46:08 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
constants.py fix format constants.py (#358) 2022-03-11 15:50:28 +08:00
core.py [polish] polish singleton and global context (#500) 2022-03-23 18:03:39 +08:00
global_variables.py [MOE] add unitest for MOE experts layout, gradient handler and kernel (#469) 2022-03-21 13:35:04 +08:00
initialize.py [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00