main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_auto_parallel\test_solver_with_resnet.py
7 3 -
A
colossalai\nn\_ops\_utils.py
46 3 -
A
.github\workflows\scripts\build_colossalai_wheel.py
20 3 -
A
colossalai\nn\_ops\embedding_bag.py
7 3 -
A
colossalai\kernel\cuda_native\csrc\kernels\transform_kernels.cu
 3 -
A
colossalai\zero\sharded_optim\sharded_optim_v2.py
73 3 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\kernels.h
10 3 -
A
colossalai\zero\sharded_model\sharded_model_v2.py
113 3 1
A
colossalai\kernel\cuda_native\csrc\scaled_masked_softmax.cpp
4 3 -
A
colossalai\utils\common.py
123 3 4
A
colossalai\gemini\chunk\chunk.py
97 3 -
A
colossalai\gemini\chunk\__init__.py
 3 -
A
colossalai\gemini\chunk\manager.py
48 3 -
A
tests\test_fx\test_meta\test_aten.py
6 3 -
A
setup.py
22 3 -
A
colossalai\gemini\chunk\search_utils.py
22 3 -
A
colossalai\fx\tracer\meta_patch\patched_module\normalization.py
4 3 -
A
colossalai\fx\tracer\meta_patch\patched_module\convolution.py
6 3 -
A
colossalai\tensor\compute_spec.py
3 3 -
A
colossalai\fx\tracer\meta_patch\patched_function\arithmetic.py
19 3 -