main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\cuda_native\csrc\kernels\cross_entropy.cu
 1 -
A
docs\conf.py
9 1 -
A
colossalai\engine\__init__.py
 1 -
A
tests\components_to_test\__init__.py
 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\cuda_util.cu
 1 -
A
tests\components_to_test\gpt.py
10 2 -
A
colossalai\amp\naive_amp\__init__.py
7 1 -
A
tests\components_to_test\no_leaf_module.py
4 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\general_kernels.cu
 1 -
A
tests\components_to_test\repeated_computed_layer.py
5 1 -
A
colossalai\engine\gradient_accumulation\__init__.py
4 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\context.h
6 1 -
A
.github\ISSUE_TEMPLATE\bug-report.yml
 1 -
A
.github\ISSUE_TEMPLATE\documentation.yml
 1 -
A
.github\workflows\assign_reviewer.yml
 1 -
A
.github\workflows\release_docker.yml
 2 -
A
colossalai\amp\naive_amp\_fp16_optimizer.py
57 1 1
A
colossalai\auto_parallel\solver\op_handler\strategy_generator.py
3 2 -
A
colossalai\engine\gradient_handler\_base_gradient_handler.py
2 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\feed_forward.h
7 1 -