LSTM-Kirigaya/ColossalAI

main

ColossalAI

Grade	Name	Complexity	Churn	Issues
A	colossalai\pipeline\rpc\_pipeline_base.py	174	4	2
A	colossalai\engine\schedule\_pipeline_schedule.py	155	2	3
B	colossalai\fx\codegen\activation_checkpoint_codegen.py	131	14	4
A	colossalai\utils\common.py	123	3	4
A	colossalai\fx\passes\algorithms\ckpt_solver_pofo.py	118	3	2
A	colossalai\nn\parallel\data_parallel.py	116	12	1
A	colossalai\fx\passes\algorithms\dynamic_programs.c	115	1	3
A	colossalai\zero\sharded_model\sharded_model_v2.py	113	3	1
A	colossalai\gemini\chunk\chunk.py	97	3	-
A	colossalai\kernel\cuda_native\csrc\scaled_upper_triang_masked_softmax.h	94	2	5
A	colossalai\kernel\cuda_native\csrc\scaled_masked_softmax.h	93	2	4
A	colossalai\nn\parallel\layers\cache_embedding\cache_mgr.py	92	23	1
A	colossalai\nn\layer\parallel_2p5d\_operation.py	84	1	-
A	colossalai\nn\layer\parallel_2d\_operation.py	81	1	-
A	colossalai\fx\passes\algorithms\ckpt_solver_rotor.py	80	14	2
A	colossalai\kernel\cuda_native\csrc\cpu_adam.cpp	80	2	3
A	tests\test_tensor\model\test_model.py	79	1	2
A	colossalai\amp\torch_amp\_grad_scaler.py	79	2	-
A	colossalai\auto_parallel\solver\strategies_constructor.py	77	14	1
A	colossalai\fx\passes\algorithms\operation.py	77	2	-