main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
extensions\pybind\flash_attention\flash_attention_dao_cuda.py
7 2 -
A
extensions\pybind\layernorm\layernorm_cuda.py
6 1 -
A
extensions\csrc\funcs\unary_functor.h
6 1 -
A
colossalai\fx\tracer\meta_patch\patched_module\convolution.py
6 5 -
A
tests\test_infer\test_continuous_batching.py
6 2 -
A
tests\test_fx\test_tracer\test_torchaudio_model\torchaudio_utils.py
6 7 -
A
colossalai\kernel\triton\rotary_cache_copy.py
6 1 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\binary_elementwise_ops.py
6 3 -
A
tests\test_fx\test_tracer\test_patched_op.py
6 4 -
A
extensions\pybind\moe\moe.cpp
6 1 -
A
colossalai\legacy\context\process_group_initializer\initializer_1d.py
6 2 -
A
colossalai\legacy\context\process_group_initializer\initializer_data.py
6 2 -
A
colossalai\legacy\context\process_group_initializer\initializer_model.py
6 2 -
A
colossalai\legacy\context\process_group_initializer\initializer_pipeline.py
6 2 -
A
colossalai\legacy\context\process_group_initializer\initializer_tensor.py
6 2 -
A
tests\test_fx\test_meta_info_prop.py
6 8 -
A
colossalai\auto_parallel\tensor_shard\node_handler\addmm_handler.py
6 2 -
A
examples\inference\benchmark_ops\benchmark_context_attn_unpad.py
6 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\default_reshape_handler.py
6 2 -
A
tests\test_analyzer\test_fx\test_symbolic_profile.py
6 4 -