main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\strategy_generator.py
33 14 -
A
tests\test_checkpoint_io\test_safetensors_async_io.py
13 5 -
A
colossalai\legacy\nn\layer\parallel_sequence\_utils.py
2 1 -
A
tests\test_checkpoint_io\test_torch_fsdp_checkpoint_io.py
17 2 -
A
.github\workflows\cuda_ext_check_before_merge.yml
 4 -
A
tests\test_cluster\test_device_mesh_manager.py
2 2 -
A
colossalai\legacy\nn\layer\utils\__init__.py
 2 -
A
tests\test_config\sample_config.py
 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\tensor_constructor_generator.py
8 2 -
A
tests\test_tensor\test_dtensor\test_layout_converter.py
5 4 -
A
tests\test_tensor\test_dtensor\test_dtensor_sharding_spec.py
3 2 -
A
tests\test_legacy\test_trainer\test_trainer_with_non_pipe_schedule.py
3 4 -
A
tests\test_legacy\test_moe\test_moe_load_balance.py
16 1 -
A
tests\test_legacy\test_layers\test_2p5d\checks_2p5d\check_operation_2p5d.py
12 4 -
A
tests\test_legacy\test_data\test_deterministic_dataloader.py
4 2 -
A
tests\test_infer\test_kernels\cuda\test_flash_decoding_attention.py
16 3 -
A
tests\test_fx\test_meta\test_meta_trace.py
4 4 -
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 -