main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\embedding_generator.py
21 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_output_handler.py
6 8 -
A
colossalai\legacy\nn\layer\colossalai_layer\linear.py
5 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_placeholder_handler.py
7 7 -
A
tests\test_shardformer\test_layer\test_dist_crossentropy.py
2 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_softmax_handler.py
8 3 -
A
colossalai\legacy\nn\layer\parallel_1d\__init__.py
 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_sum_handler.py
15 3 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\getitem_generator.py
27 7 -
A
tests\test_shardformer\test_layer\test_dropout.py
10 3 -
A
tests\test_shardformer\test_layer\test_dist_log_prob.py
0 1 -
A
tests\test_legacy\test_data\test_data_parallel_sampler.py
4 2 -
A
tests\test_lazy\lazy_init_utils.py
15 8 -
A
tests\test_infer\test_kernels\triton\test_decoding_attn.py
14 3 -
A
tests\test_infer\test_drafter.py
4 1 -
A
tests\test_fx\test_pipeline\test_hf_model\test_gpt.py
2 5 -
A
tests\test_booster\test_plugin\test_torch_ddp_plugin.py
19 8 -
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 -