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

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

${ noResults }

ColossalAI/tests/test_zero/test_gemini

History

Hongxin Liu ae02d4e4f7 [bf16] add bf16 support (#3882) ... * [bf16] add bf16 support for fused adam (#3844) * [bf16] fused adam kernel support bf16 * [test] update fused adam kernel test * [test] update fused adam test * [bf16] cpu adam and hybrid adam optimizers support bf16 (#3860) * [bf16] implement mixed precision mixin and add bf16 support for low level zero (#3869) * [bf16] add mixed precision mixin * [bf16] low level zero optim support bf16 * [text] update low level zero test * [text] fix low level zero grad acc test * [bf16] add bf16 support for gemini (#3872) * [bf16] gemini support bf16 * [test] update gemini bf16 test * [doc] update gemini docstring * [bf16] add bf16 support for plugins (#3877) * [bf16] add bf16 support for legacy zero (#3879) * [zero] init context support bf16 * [zero] legacy zero support bf16 * [test] add zero bf16 test * [doc] add bf16 related docstring for legacy zero		1 year ago
..
test_chunk_mgrv2.py	[test] refactor tests with spawn (#3452)	2 years ago
test_chunkv2.py	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_fwd_bwd.py	[gemini] accelerate inference (#3641)	2 years ago
test_gemini_use_rmt.py	[test] refactor tests with spawn (#3452)	2 years ago
test_get_torch_model.py	[test] refactor tests with spawn (#3452)	2 years ago
test_grad_clip.py	[test] refactor tests with spawn (#3452)	2 years ago
test_inference.py	[test] refactor tests with spawn (#3452)	2 years ago
test_optim.py	[bf16] add bf16 support (#3882)	1 year ago
test_runtime_mem_tracer.py	[test] refactor tests with spawn (#3452)	2 years ago
test_search.py	[test] refactor tests with spawn (#3452)	2 years ago
test_zeroddp_state_dict.py	[test] refactor tests with spawn (#3452)	2 years ago
test_zeroddp_state_dict_shard.py	[booster] gemini plugin support shard checkpoint (#3610)	2 years ago
test_zerooptim_state_dict.py	[test] refactor tests with spawn (#3452)	2 years ago