main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\offload\amp_optimizer.py
22 5 -
A
colossalai\auto_parallel\offload\base_offload_module.py
25 5 -
A
colossalai\auto_parallel\offload\mem_optimize.py
4 3 -
A
colossalai\auto_parallel\offload\region.py
17 3 -
A
colossalai\auto_parallel\offload\region_manager.py
82 2 -
A
colossalai\auto_parallel\offload\runtime.py
54 3 -
A
colossalai\auto_parallel\offload\solver.py
72 4 -
A
colossalai\auto_parallel\offload\training_simulator.py
72 3 -
A
colossalai\auto_parallel\offload\util.py
22 3 -
A
colossalai\auto_parallel\passes\comm_metainfo_pass.py
12 3 -
A
colossalai\auto_parallel\passes\constants.py
 2 -
A
colossalai\auto_parallel\passes\meta_info_prop.py
32 5 -
A
colossalai\auto_parallel\passes\runtime_apply_pass.py
53 13 -
A
colossalai\auto_parallel\passes\runtime_preparation_pass.py
52 29 -
A
colossalai\auto_parallel\tensor_shard\constants.py
 4 -
A
colossalai\auto_parallel\tensor_shard\initialize.py
34 11 -
A
colossalai\auto_parallel\tensor_shard\node_handler\__init__.py
 19 -
A
colossalai\auto_parallel\tensor_shard\node_handler\addmm_handler.py
6 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\batch_norm_handler.py
3 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\binary_elementwise_handler.py
8 7 -