main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_infer\test_models\test_attention.py
4 1 -
A
tests\test_infer\test_kvcache_manager.py
14 4 -
A
tests\test_infer\test_kernels\triton\test_xine_copy.py
5 1 -
A
tests\test_infer\test_kernels\triton\test_rotary_embdding_unpad.py
4 2 -
A
tests\test_infer\test_kernels\triton\test_rmsnorm_triton.py
1 1 -
A
tests\test_infer\test_kernels\triton\test_kvcache_copy.py
12 1 -
A
tests\test_infer\test_kernels\triton\test_fused_rotary_embedding.py
1 2 -
A
tests\test_infer\test_kernels\triton\test_decoding_attn.py
14 3 -
A
tests\test_infer\test_kernels\triton\test_context_attn_unpad.py
12 2 -
A
tests\test_infer\test_kernels\triton\kernel_utils.py
35 1 -
A
tests\test_infer\test_kernels\cuda\test_silu_and_mul.py
2 1 -
A
tests\test_infer\test_kernels\cuda\test_rotary_embdding_unpad.py
5 2 -
A
tests\test_infer\test_kernels\cuda\test_rms_layernorm.py
1 1 -
A
tests\test_infer\test_kernels\cuda\test_kv_cache_memcpy.py
9 2 -
A
tests\test_infer\test_kernels\cuda\test_get_cos_and_sin.py
2 1 -
A
tests\test_infer\test_kernels\cuda\test_flash_decoding_attention.py
16 3 -
A
tests\test_infer\test_kernels\cuda\test_convert_fp8.py
1 1 -
A
tests\test_infer\test_inference_engine.py
 1 -
A
tests\test_infer\test_drafter.py
4 1 -
A
tests\test_infer\test_cuda_graph.py
8 1 -