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.

flybird1111 38b792aab2 [coloattention] fix import error (#4380) ... fixed an import error		1 year ago
..
components_to_test	…
kit	[shardformer] added embedding gradient check (#4124)	1 year ago
test_amp	…
test_analyzer	…
test_auto_parallel	[gemini] fix argument naming during chunk configuration searching	1 year ago
test_autochunk	[test] fixed tests failed due to dtensor change (#4082)	1 year ago
test_booster	[zero] refactor low level zero for shard evenly (#4030)	1 year ago
test_checkpoint_io	[zero] support shard optimizer state dict of zero (#4194)	1 year ago
test_cluster	…
test_comm	…
test_config	…
test_context	…
test_data	…
test_data_pipeline_tensor_parallel	…
test_ddp	…
test_device	[format] applied code formatting on changed files in pull request 4152 (#4157)	1 year ago
test_engine	…
test_fx	[shardformer] shardformer support opt models (#4091)	1 year ago
test_kernels	[Kernels] added triton-implemented of self attention for colossal-ai (#4241)	1 year ago
test_layers	…
test_lazy	[test] remove useless tests (#4359)	1 year ago
test_moe	…
test_ops	…
test_optimizer	[bf16] add bf16 support (#3882)	1 year ago
test_pipeline	…
test_shardformer	[format] applied code formatting on changed files in pull request 4152 (#4157)	1 year ago
test_tensor	[test] fixed tests failed due to dtensor change (#4082)	1 year ago
test_trainer	…
test_utils	[coloattention] fix import error (#4380)	1 year ago
test_zero	[test] remove useless tests (#4359)	1 year ago
__init__.py	…