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
Cuiqing Li (李崔卿) bce919708f
[Kernels]added flash-decoidng of triton (#5063)
1 year ago
..
_C [setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago
_analyzer [misc] update pre-commit and run all files (#4752) 1 year ago
amp [feature] Add clip_grad_norm for hybrid_parallel_plugin (#4837) 1 year ago
auto_parallel [misc] update pre-commit and run all files (#4752) 1 year ago
autochunk [misc] update pre-commit and run all files (#4752) 1 year ago
booster [gemini] gemini support extra-dp (#5043) 1 year 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 [gemini] gemini support tensor parallelism. (#4942) 1 year ago
context [moe] merge moe into main (#4978) 1 year ago
device [misc] update pre-commit and run all files (#4752) 1 year ago
fx [misc] update pre-commit and run all files (#4752) 1 year ago
inference [Kernels]added flash-decoidng of triton (#5063) 1 year ago
interface [lazy] support from_pretrained (#4801) 1 year ago
kernel [Kernels]added flash-decoidng of triton (#5063) 1 year ago
lazy [doc] add lazy init docs (#4808) 1 year ago
legacy [inference] Refactor inference architecture (#5057) 1 year ago
logging [misc] update pre-commit and run all files (#4752) 1 year ago
moe [hotfix]: modify create_ep_hierarchical_group and add test (#5032) 1 year ago
nn [moe] merge moe into main (#4978) 1 year ago
pipeline [inference] Refactor inference architecture (#5057) 1 year ago
shardformer [inference] Refactor inference architecture (#5057) 1 year ago
tensor [hotfix]: modify create_ep_hierarchical_group and add test (#5032) 1 year ago
testing [test] merge old components to test to model zoo (#4945) 1 year ago
utils [moe] merge moe into main (#4978) 1 year ago
zero [gemini] gemini support extra-dp (#5043) 1 year ago
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago
initialize.py [misc] update pre-commit and run all files (#4752) 1 year ago