Frank Lee
fc5101f24c
Merge pull request #401 from hpcaitech/develop
3 years ago
Frank Lee
fc2fd0abe5
Merge pull request #400 from hpcaitech/hotfix/readme
...
fixed broken badge link
3 years ago
Frank Lee
6d3a4f51bf
fixed broken badge link
3 years ago
HELSON
7c079d9c33
[hotfix] fixed bugs in ShardStrategy and PcieProfiler ( #394 )
3 years ago
Frank Lee
1e4bf85cdb
fixed bug in activation checkpointing test ( #387 )
3 years ago
Jiarui Fang
3af13a2c3e
[zero] polish ShardedOptimV2 unittest ( #385 )
...
* place params on cpu after zero init context
* polish code
* bucketzed cpu gpu tensor transter
* find a bug in sharded optim unittest
* add offload unittest for ShardedOptimV2.
* polish code and make it more robust
3 years ago
binmakeswell
ce7b2c9ae3
update README and images path ( #384 )
3 years ago
ScalableEKNN
2fcd4f38ee
fix format ( #379 )
3 years ago
Jiang Zhuo
5a4a3b77d9
fix format ( #376 )
3 years ago
lucasliunju
ce886a9062
fix format ( #374 )
3 years ago
Frank Lee
526a318032
[unit test] Refactored test cases with component func ( #339 )
...
* refactored test with component func
* fixed bug
3 years ago
LuGY
de46450461
Added activation offload ( #331 )
...
* Added activation offload
* Fixed the import bug, used the pytest
3 years ago
Jiarui Fang
272ebfb57d
[bug] shard param during initializing the ShardedModelV2 ( #381 )
3 years ago
HELSON
8c18eb0998
[profiler] Fixed bugs in CommProfiler and PcieProfiler ( #377 )
3 years ago
Jiarui Fang
b5f43acee3
[zero] find miss code ( #378 )
3 years ago
Jiarui Fang
6b6002962a
[zero] zero init context collect numel of model ( #375 )
3 years ago
HELSON
1ed7c24c02
Added PCIE profiler to dectect data transmission ( #373 )
3 years ago
jiaruifang
d9217e1960
Revert "[zero] bucketized tensor cpu gpu copy ( #368 )"
...
This reverts commit bef05489b6
.
3 years ago
Xue Fuzhao
a8cd5e8e81
Update README-zh-Hans.md ( #367 )
...
Fuzhao updated
3 years ago
Shen Chenhui
1c88dd43e2
Fix/format ( #366 )
3 years ago
Ziheng Qin
0db43fa995
fix format ( #364 )
3 years ago
RichardoLuo
8539898ec6
flake8 style change ( #363 )
3 years ago
Kai Wang (Victor Kai)
53bb3bcc0a
fix format ( #362 )
3 years ago
ziyu huang
a77d73f22b
fix format parallel_context.py ( #359 )
...
Co-authored-by: huangziyu <202476410arsmart@gmail.com>
3 years ago
Zangwei
c695369af0
fix format constants.py ( #358 )
3 years ago
Yuer867
4a0f8c2c50
fix format parallel_2p5d ( #357 )
3 years ago
Liang Bowen
7eb87f516d
flake8 style ( #352 )
3 years ago
Xu Kai
54ee8d1254
Fix/format colossalai/engine/paramhooks/( #350 )
3 years ago
Maruyama_Aya
e83970e3dc
fix format ColossalAI\colossalai\context\process_group_initializer
3 years ago
yuxuan-lou
3b88eb2259
Flake8 code restyle
3 years ago
xyupeng
af801cb4df
fix format setup.py ( #343 )
3 years ago
xuqifan897
148207048e
Qifan formated file ColossalAI\colossalai\nn\layer\parallel_1d\layers.py ( #342 )
3 years ago
Cautiousss
3a51d909af
fix format ( #332 )
...
Co-authored-by: 何晓昕 <cautious@r-205-106-25-172.comp.nus.edu.sg>
3 years ago
DouJS
cbb6436ff0
fix format for dir-[parallel_3d] ( #333 )
3 years ago
ExtremeViscent
eaac03ae1d
[formart] format fixed for kernel\cuda_native codes ( #335 )
3 years ago
Jiarui Fang
00670c870e
[zero] bucketized tensor cpu gpu copy ( #368 )
3 years ago
Jiarui Fang
44e4891f57
[zero] able to place params on cpu after zero init context ( #365 )
...
* place params on cpu after zero init context
* polish code
3 years ago
ver217
b66f3b994c
increase the timeout limit in CI temporarily
3 years ago
ver217
52d055119b
increase the timeout limit in CI temporarily
3 years ago
ver217
253e54d98a
fix grad shape
3 years ago
Jiarui Fang
ea2872073f
[zero] global model data memory tracer ( #360 )
3 years ago
Jiarui Fang
cb34cd384d
[test] polish zero related unitest ( #351 )
3 years ago
HELSON
534e0bb118
Fixed import bug for no-tensorboard environment ( #354 )
3 years ago
HELSON
c57e089824
[profile] added example for ProfilerContext ( #349 )
3 years ago
ver217
532ae79cb0
add test sharded optim with cpu adam ( #347 )
3 years ago
Jiarui Fang
10e2826426
move async memory to an individual directory ( #345 )
3 years ago
HELSON
425bb0df3f
Added Profiler Context to manage all profilers ( #340 )
3 years ago
ver217
d0ae0f2215
[zero] update sharded optim v2 ( #334 )
3 years ago
ver217
2b8cddd40e
skip bert in test engine
3 years ago
ver217
d41a9f12c6
install transformers in CI
3 years ago