Commit Graph

570 Commits (1d625fcd36783d866b3d1d1b208e0f2291b545d3)
 

Author SHA1 Message Date
Jiarui Fang 193dc8dacb
[refactor] refactor the memory utils (#715)
3 years ago
HELSON dbd96fe90a
[zero] check whether gradients have inf and nan in gpu (#712)
3 years ago
ver217 715b86eadd
[hotfix] fix stm cuda model data size (#710)
3 years ago
LuGY 140263a394
[hotfix]fixed bugs of assigning grad states to non leaf nodes (#711)
3 years ago
Frank Lee eda30a058e
[compatibility] fixed tensor parallel compatibility with torch 1.9 (#700)
3 years ago
HELSON a9b8300d54
[zero] improve adaptability for not-shard parameters (#708)
3 years ago
ver217 ab8c6b4a0e
[zero] refactor memstats collector (#706)
3 years ago
アマデウス 3fc8a204dc
[]Corrected 3d vocab parallel embedding (#707)
3 years ago
HELSON ee112fe1da
[zero] adapt zero hooks for unsharded module (#699)
3 years ago
binmakeswell 896ade15d6
add PaLM link (#704) (#705)
3 years ago
binmakeswell 270157e9e7
add PaLM link (#704)
3 years ago
ver217 3c9cd5bb5e
[zero] stateful tensor manager (#687)
3 years ago
ver217 70e8dd418b
[hotfix] update requirements-test (#701)
3 years ago
Frank Lee 1ae94ea85a
[ci] remove ipc config for rootless docker (#694)
3 years ago
github-actions[bot] d878d843ad
Automated submodule synchronization (#695)
3 years ago
github-actions[bot] d50cdabbc9
Automated submodule synchronization (#556)
3 years ago
Frank Lee dbe8e030fb
[ci] added missing field in workflow (#692)
3 years ago
Frank Lee 0372ed7951
[ci] update workflow trigger condition and support options (#691)
3 years ago
HELSON d7ecaf362b
[zero] fix init bugs in zero context (#686)
3 years ago
YuliangLiu0306 0ed7042f42
[pipeline] refactor pipeline (#679)
3 years ago
Frank Lee eace69387d
[ci] fixed compatibility workflow (#678)
3 years ago
Jiarui Fang 59bf2dc590
[zero] initialize a stateful tensor manager (#614)
3 years ago
Frank Lee cc236916c6
[ci] replace the dngc ocker image with self-built pytorch image (#672)
3 years ago
ver217 03e1d35931
[release] update version (#673)
3 years ago
encmps 79ccfa4310 [NFC] polish colossalai/kernel/cuda_native/csrc/multi_tensor_adam.cu code style (#667)
3 years ago
lucasliunju e4bcff9b0f [NFC] polish colossalai/builder/builder.py code style (#662)
3 years ago
shenggan 331683bf82 [NFC] polish colossalai/kernel/cuda_native/csrc/layer_norm_cuda_kernel.cu code style (#661)
3 years ago
FredHuang99 c336cd3066 [NFC] polish colossalai/communication/utils.py code style (#656)
3 years ago
MaxT 5ab9a71299 [NFC] polish colossalai/kernel/cuda_native/csrc/moe_cuda.cpp code style (#642)
3 years ago
Xue Fuzhao 10afec728f [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h code style (#641)
3 years ago
Cautiousss 055d0270c8 [NFC] polish colossalai/context/process_group_initializer/initializer_sequence.py colossalai/context/process_group_initializer initializer_tensor.py code style (#639)
3 years ago
Ziheng Qin c7c224ee17 [NFC] polish colossalai/builder/pipeline.py code style (#638)
3 years ago
Sze-qq 10591ecdf9 [NFC] polish colossalai/kernel/cuda_native/csrc/cpu_adam.cpp code style (#636)
3 years ago
Wangbo Zhao 6fcb381801 [NFC] polish colossalai/kernel/cuda_native/csrc/multi_tensor_l2norm_kernel.cu code style (#635)
3 years ago
ExtremeViscent 8a5d526e95 [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu and cross_entropy.cu code style (#634)
3 years ago
RichardoLuo ad1e7ab2b2 '[NFC] polish <colossalai/engine/_base_engine.py> code style' (#631)
3 years ago
Zangwei 2e11853d04 [NFC] polish colossalai/communication/ring.py code style (#630)
3 years ago
puck_WCR 01cc941e1d [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu code stype (#629)
3 years ago
superhao1995 c1bed0d998 [NFC] polish colossalai/kernel/cuda_native/csrc/multi_tensor_lamb.cu code stype (#628)
3 years ago
Jiang Zhuo 0a96338b13 [NFC] polish <colossalai/context/process_group_initializer/initializer_data.py> code stype (#626)
3 years ago
ziyu huang 701bad439b [NFC] polish colossalai/context/process_group_initializer/process_group_initializer.py code stype (#617)
3 years ago
Shawn-Kong db54419409 fix format (#613)
3 years ago
Yuer867 5ecef13c16 fix format (#611)
3 years ago
xyupeng d3d5bedc65 fix format (#607)
3 years ago
xuqifan897 f2d2a1597a fix format (#608)
3 years ago
doubleHU f2da21a827 fix format (#586)
3 years ago
fanjinfucool ffad81e1d1 fix format (#585)
3 years ago
binmakeswell 6582aedc94 fix format (#583)
3 years ago
DouJS f08fc17f2b block_reduce.h fix format (#581)
3 years ago
Maruyama_Aya d2dc6049b5 fix format (#580)
3 years ago