main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\Colossal-LLaMA\colossal_llama\model\init_model.py
6 1 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_pooling_metainfo.py
4 3 -
A
colossalai\legacy\moe\utils.py
42 1 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_where_metainfo.py
1 4 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\__init__.py
 9 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_addbmm_handler.py
17 3 -
A
colossalai\legacy\nn\_ops\__init__.py
 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_batch_norm_handler.py
4 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_getitem_handler.py
9 3 -
A
tests\test_optimizer\test_dist_adafactor.py
32 2 -
A
tests\test_optimizer\test_adam_optim.py
10 2 -
A
tests\test_infer\test_models\test_baichuan.py
 2 -
A
tests\test_infer\test_kernels\cuda\test_get_cos_and_sin.py
2 1 -
A
tests\test_fx\test_tracer\test_torchrec_model\test_dlrm_model.py
11 3 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_bert.py
3 13 -
A
tests\test_fp8\test_fp8_cast.py
2 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_solver_with_resnet_v2.py
5 9 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -