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.

Branch: main

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

${ noResults }

ColossalAI/tests/test_auto_parallel/test_tensor_shard

History

Hongxin Liu 7f8b16635b [misc] refactor launch API and tensor constructor (#5666) ... * [misc] remove config arg from initialize * [misc] remove old tensor contrusctor * [plugin] add npu support for ddp * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [devops] fix doc test ci * [test] fix test launch * [doc] update launch doc --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>		7 months ago
..
test_gpt	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_metainfo	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_node_handler	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
__init__.py	[autoparallel] handled illegal sharding strategy (#1728)	2 years ago
test_bias_addition_forward.py	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_broadcast.py	[misc] update pre-commit and run all files (#4752)	1 year ago
test_checkpoint.py	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_compatibility_with_ddp.py	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_compatibility_with_gemini.py	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_find_repeat_block.py	[misc] update pre-commit and run all files (#4752)	1 year ago
test_liveness_analysis.py	[misc] update pre-commit and run all files (#4752)	1 year ago
test_solver_with_resnet_v2.py	[misc] update pre-commit and run all files (#4752)	1 year ago