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: 8e08c27e19

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 '8e08c27e19'

${ noResults }

ColossalAI/tests/test_zero/test_gemini

History

botbw 80c3c8789b [Test/CI] remove test cases to reduce CI duration (#5753) ... * [test] smaller gpt2 test case * [test] reduce test cases: tests/test_zero/test_gemini/test_zeroddp_state_dict.py * [test] reduce test cases: tests/test_zero/test_gemini/test_grad_accum.py * [test] reduce test cases tests/test_zero/test_gemini/test_optim.py * Revert "[test] smaller gpt2 test case" Some tests might depend on the size of model (num of chunks) This reverts commit df705a5210. * [test] reduce test cases: tests/test_checkpoint_io/test_gemini_checkpoint_io.py * [CI] smaller test model for two mwo the two modifid cases * [CI] hardcode gpt model for tests/test_zero/test_gemini/test_search.py since we need a fixed answer there		6 months ago
..
test_chunk_mgrv2.py	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_chunkv2.py	[gemini] async grad chunk reduce (all-reduce&reduce-scatter) (#5713)	6 months ago
test_gemini_use_rmt.py	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_grad_accum.py	[Test/CI] remove test cases to reduce CI duration (#5753)	6 months ago
test_grad_clip.py	Merge branch 'main' of github.com:hpcaitech/ColossalAI into prefetch	6 months ago
test_inference.py	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_optim.py	[Test/CI] remove test cases to reduce CI duration (#5753)	6 months ago
test_runtime_mem_tracer.py	…
test_search.py	[Test/CI] remove test cases to reduce CI duration (#5753)	6 months ago
test_zeroddp_state_dict.py	[Test/CI] remove test cases to reduce CI duration (#5753)	6 months ago
test_zerooptim_state_dict.py	[chore] remove unnecessary test & changes	6 months ago