You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai
ver217 148469348a
Merge branch 'main' into sync/npu
10 months ago
..
_C
_analyzer [misc] update pre-commit and run all files (#4752) 1 year ago
accelerator [hotfix] removed unused flag (#5242) 11 months ago
amp [npu] change device to accelerator api (#5239) 11 months ago
auto_parallel [npu] change device to accelerator api (#5239) 11 months ago
autochunk [misc] update pre-commit and run all files (#4752) 1 year ago
booster Merge branch 'main' into sync/npu 10 months ago
checkpoint_io [pipeline,shardformer] Fix p2p efficiency in pipeline, allow skipping loading weight not in weight_map when `strict=False`, fix llama flash attention forward, add flop estimation by megatron in llama benchmark (#5017) 1 year ago
cli [bug] Fix the version check bug in colossalai run when generating the cmd. (#4713) 1 year ago
cluster fix-test (#5210) 11 months ago
context [moe] merge moe into main (#4978) 1 year ago
device [npu] add npu support for hybrid plugin and llama (#5090) 1 year ago
fx [misc] update pre-commit and run all files (#4752) 1 year ago
inference [Hotfix] Fix model policy matching strategy in ShardFormer (#5064) 1 year ago
interface [lazy] support from_pretrained (#4801) 1 year ago
kernel [npu] change device to accelerator api (#5239) 11 months ago
lazy [doc] add lazy init docs (#4808) 1 year ago
legacy Merge branch 'main' into sync/npu 10 months ago
logging [misc] update pre-commit and run all files (#4752) 1 year ago
moe [npu] change device to accelerator api (#5239) 11 months ago
nn [npu] use extension for op builder (#5172) 11 months ago
pipeline Merge branch 'main' into sync/npu 10 months ago
shardformer Merge branch 'main' into sync/npu 10 months ago
tensor fix (#5158) 12 months ago
testing [npu] change device to accelerator api (#5239) 11 months ago
utils Merge branch 'main' into sync/npu 10 months ago
zero Merge branch 'main' into sync/npu 10 months ago
__init__.py [accelerator] init the accelerator module (#5129) 1 year ago
initialize.py [npu] change device to accelerator api (#5239) 11 months ago