id: 1
unit: def run_flash_infer()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 43 LOC
McCabe index: 4
number of parameters: 14

id: 2
unit: def compare_ab()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 3
number of parameters: 11

id: 3
unit: def _mla_attn_kernel()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 1
number of parameters: 0

id: 4
unit: def run_torch_mla()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 3
number of parameters: 14

id: 5
unit: def compare_a()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 1
number of parameters: 10

id: 6
unit: def scaled_dot_product_attention()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 6

id: 7
unit: def run_flash_mla_triton()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 2
number of parameters: 14

id: 8
unit: def run_flash_mla()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 14

id: 9
unit: def mla_decode_triton()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 10
unit: def _mla_attn()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 11
unit: def _mla_softmax_reducev_kernel()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 12
unit: def flash_mla_with_kvcache()
file: flash_mla/flash_mla_interface.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 13
unit: def get_args()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 14
unit: def get_features_args()
file: setup.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 15
unit: def _mla_softmax_reducev()
file: benchmark/bench_flash_mla.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 16
unit: def parse_args()
file: benchmark/visualize.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 17
unit: def get_mla_metadata()
file: flash_mla/flash_mla_interface.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 18
unit: def append_nvcc_threads()
file: setup.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1