github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-24 13:03:00 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b82d60be0294bf20058b60ca86fbd5cdc761bb21
ColossalAI/tests
History
Jiarui Fang 569357fea0 add pytorch hooks (#179) 
* add pytorch hooks
fix #175

* remove licenses in src code

* add gpu memory tracer

* replacing print with logger in ophooks.
2022-01-25 22:20:54 +08:00
..
test_comm
Hotfix/Colossalai layers (#92)
2021-12-29 23:32:10 +08:00
test_config
add pytorch hooks (#179)
2022-01-25 22:20:54 +08:00
test_context
Optimize pipeline schedule (#94)
2021-12-30 15:56:46 +08:00
test_data
added CI for unit testing (#69)
2021-12-16 10:32:08 +08:00
test_data_pipeline_tensor_parallel
Optimize pipeline schedule (#94)
2021-12-30 15:56:46 +08:00
test_engine/test_engine
Hotfix/Colossalai layers (#92)
2021-12-29 23:32:10 +08:00
test_layers
adapted for sequence parallel (#163)
2022-01-20 13:44:51 +08:00
test_trainer
pipeline last stage supports multi output (#151)
2022-01-17 15:57:47 +08:00
test_utils
Hotfix/Colossalai layers (#92)
2021-12-29 23:32:10 +08:00
test_zero_data_parallel
Hotfix/Colossalai layers (#92)
2021-12-29 23:32:10 +08:00
test_zero_tensor_parallel
Hotfix/Colossalai layers (#92)
2021-12-29 23:32:10 +08:00