mirror of https://github.com/hpcaitech/ColossalAI
19e1a5cf16
* [feature] refactor colo attention (#5462) * [extension] update api * [feature] add colo attention * [feature] update sdpa * [feature] update npu attention * [feature] update flash-attn * [test] add flash attn test * [test] update flash attn test * [shardformer] update modeling to fit colo attention (#5465) * [misc] refactor folder structure * [shardformer] update llama flash-attn * [shardformer] fix llama policy * [devops] update tensornvme install * [test] update llama test * [shardformer] update colo attn kernel dispatch * [shardformer] update blip2 * [shardformer] update chatglm * [shardformer] update gpt2 * [shardformer] update gptj * [shardformer] update opt * [shardformer] update vit * [shardformer] update colo attention mask prep * [shardformer] update whisper * [test] fix shardformer tests (#5514) * [test] fix shardformer tests * [test] fix shardformer tests |
||
|---|---|---|
| .. | ||
| kit | ||
| test_analyzer | ||
| test_auto_parallel | ||
| test_autochunk | ||
| test_booster | ||
| test_checkpoint_io | ||
| test_cluster | ||
| test_config | ||
| test_device | ||
| test_fx | ||
| test_gptq | ||
| test_infer | ||
| test_lazy | ||
| test_legacy | ||
| test_moe | ||
| test_optimizer | ||
| test_pipeline | ||
| test_shardformer | ||
| test_smoothquant | ||
| test_tensor | ||
| test_zero | ||
| __init__.py | ||