Making large AI models cheaper, faster and more accessible

aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism

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.

Tree: 2a718c8be8

ColossalChat

ckpt

cloud/coati

colossalchat

colossalchat_upgrade

dev/gpt2_metainfo_patch

dev/zero-offload

develop

feat/moe

feat/online-serving

feat/speculative-decoding

feature/2-stage

feature/async-io

feature/colossal-infer

feature/elixir

feature/inference-refactor

feature/lora

feature/stable-diffusion

feature/zerobubble

fix-setup

fix/format

flybird11111-patch-1

hotfix/example_test

hotfix/kernel_build_before_load

llama3

main

moe_sp

pre-commit-ci-update-config

refactor/inference

supercooledith-patch-1

support-npu

v0.0.1-beta

v0.0.2

v0.1.0

v0.1.1

v0.1.10

v0.1.11rc1

v0.1.11rc2

v0.1.11rc3

v0.1.11rc4

v0.1.11rc5

v0.1.12

v0.1.13

v0.1.2

v0.1.3

v0.1.4

v0.1.5

v0.1.6

v0.1.7

v0.1.8

v0.1.9

v0.2.0

v0.2.1

v0.2.2

v0.2.3

v0.2.4

v0.2.5

v0.2.6

v0.2.7

v0.2.8

v0.3.0

v0.3.1

v0.3.2

v0.3.3

v0.3.4

v0.3.5

v0.3.6

v0.3.7

v0.3.8

v0.3.9

v0.4.0

v0.4.1

v0.4.2

v0.4.3

v0.4.4

v0.4.5

v0.4.6

Branches Tags

${ item.name }

Create tag ${ searchTerm }

Create branch ${ searchTerm }

from '2a718c8be8'

${ noResults }

ColossalAI/colossalai

History

yuehuayingxueluo 2a718c8be8 Optimized the execution interval time between cuda kernels caused by view and memcopy (#5390) ... * opt_view_and_memcopy * fix bugs in ci * fix ci bugs * update benchmark scripts * fix ci bugs		9 months 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
accelerator	[accelerator] fixed npu api	10 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	Optimized the execution interval time between cuda kernels caused by view and memcopy (#5390)	9 months ago
interface	[lazy] support from_pretrained (#4801)	1 year ago
kernel	Optimized the execution interval time between cuda kernels caused by view and memcopy (#5390)	9 months ago
lazy	[doc] add lazy init docs (#4808)	1 year ago
legacy	merge commit	10 months ago
logging	[misc] update pre-commit and run all files (#4752)	1 year ago
moe	Merge pull request #5310 from hpcaitech/feature/npu	10 months ago
nn	[feat] refactored extension module (#5298)	10 months ago
pipeline	[feat] refactored extension module (#5298)	10 months ago
shardformer	fix typo change dosen't to doesn't (#5308)	10 months ago
tensor	fix some typo (#5307)	10 months ago
testing	[npu] change device to accelerator api (#5239)	11 months ago
utils	Merge pull request #5310 from hpcaitech/feature/npu	10 months ago
zero	Merge pull request #5310 from hpcaitech/feature/npu	10 months ago
__init__.py	[accelerator] init the accelerator module (#5129)	12 months ago
initialize.py	[npu] change device to accelerator api (#5239)	11 months ago