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: ad2cf58f50

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 'ad2cf58f50'

${ noResults }

ColossalAI/tests

History

Hongxin Liu 5452df63c5 [plugin] torch ddp plugin supports sharded model checkpoint (#3775) ... * [plugin] torch ddp plugin add save sharded model * [test] fix torch ddp ckpt io test * [test] fix torch ddp ckpt io test * [test] fix low level zero plugin test * [test] fix low level zero plugin test * [test] add debug info * [test] add debug info * [test] add debug info * [test] add debug info * [test] add debug info * [test] fix low level zero plugin test * [test] fix low level zero plugin test * [test] remove debug info		2 years ago
..
components_to_test	[CI] fix typo with tests/ etc. (#3727)	2 years ago
kit	…
test_amp	…
test_analyzer	[devops] update torch version of CI (#3725)	2 years ago
test_auto_parallel	…
test_autochunk	[devops] update torch version of CI (#3725)	2 years ago
test_booster	[plugin] torch ddp plugin supports sharded model checkpoint (#3775)	2 years ago
test_checkpoint_io	[plugin] torch ddp plugin supports sharded model checkpoint (#3775)	2 years ago
test_cluster	[devops] update torch version of CI (#3725)	2 years ago
test_comm	…
test_config	…
test_context	…
test_data	…
test_data_pipeline_tensor_parallel	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_ddp	…
test_device	…
test_engine	…
test_fx	[devops] update torch version of CI (#3725)	2 years ago
test_layers	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_moe	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_ops	…
test_optimizer	…
test_pipeline	…
test_tensor	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_trainer	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_utils	[devops] update torch version of CI (#3725)	2 years ago
test_zero	[CI] fix typo with tests/ etc. (#3727)	2 years ago
__init__.py	…