main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
examples\tutorial\auto_parallel\auto_ckpt_solver_test.py
9 3 -
A
examples\language\opt\data.py
9 2 -
A
tests\test_analyzer\test_fx\test_mod_dir.py
9 3 -
A
tests\test_zero\test_gemini\test_runtime_mem_tracer.py
9 5 -
A
tests\test_shardformer\test_with_torch_ddp.py
9 4 -
A
tests\test_legacy\test_amp\test_naive_fp16.py
9 4 -
A
examples\tutorial\sequence_parallel\model\layers\head.py
9 4 -
A
colossalai\legacy\zero\shard_utils\tensor_shard_strategy.py
9 3 -
A
examples\tutorial\sequence_parallel\model\layers\bert_layer.py
9 4 -
A
colossalai\cluster\process_group_manager.py
9 2 -
A
applications\ColossalMoE\utils.py
9 1 1
A
tests\test_legacy\test_amp\test_torch_fp16.py
9 4 -
A
tests\test_infer\test_kernels\cuda\test_kv_cache_memcpy.py
9 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_shard_option.py
9 5 -
A
examples\language\gpt\titans\dataset\webtext.py
8 3 -
A
colossalai\fx\tracer\bias_addition_patch\patched_bias_addition_function\addbmm.py
8 2 -
A
colossalai\fx\tracer\bias_addition_patch\patched_bias_addition_module\bias_addition_module.py
8 3 -
A
applications\Colossal-LLaMA\colossal_llama\utils\utils.py
8 1 -
A
colossalai\zero\low_level\zero_hook.py
8 1 -
A
colossalai\inference\core\base_engine.py
8 1 -