main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_legacy\test_data\test_data_parallel_sampler.py
4 2 -
A
tests\test_lazy\test_ops.py
4 1 -
A
tests\test_fx\test_parallel_1d.py
4 4 -
A
colossalai\fx\profiler\experimental\profiler_module\normalization.py
4 3 -
A
colossalai\fx\profiler\experimental\registry.py
4 2 -
A
colossalai\utils\multi_tensor_apply\multi_tensor_apply.py
5 6 -
A
colossalai\legacy\nn\parallel\layers\linear.py
5 3 -
A
colossalai\inference\graph_runner.py
5 1 -
A
colossalai\shardformer\shard\utils.py
5 1 -
A
extensions\pybind\softmax\scaled_upper_triangle_masked_softmax_cuda.py
5 1 -
A
colossalai\inference\modeling\layers\baichuan_tp_linear.py
5 3 -
A
tests\kit\model_zoo\transformers\falcon.py
5 1 -
A
tests\test_auto_parallel\test_tensor_shard\test_solver_with_resnet_v2.py
5 9 -
A
examples\language\gpt\experiments\auto_offload\model_zoo.py
5 2 -
A
tests\test_checkpoint_io\test_torch_ddp_checkpoint_io.py
5 3 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\base.py
5 2 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\bf16.py
5 1 -
A
colossalai\legacy\amp\__init__.py
5 2 -
A
tests\kit\model_zoo\transformers\bloom.py
5 6 -
A
extensions\pybind\softmax\scaled_masked_softmax_cuda.py
5 1 -