path	# lines of code
csrc/kernels/utils.h	25
csrc/kernels/get_mla_metadata.cu	73
csrc/kernels/splitkv_mla.cu	1230
csrc/kernels/mla_combine.h	4
csrc/kernels/config.h	8
csrc/kernels/params.h	46
csrc/kernels/get_mla_metadata.h	3
csrc/kernels/splitkv_mla.h	4
csrc/kernels/mla_combine.cu	179
csrc/kernels/traits.h	87
csrc/flash_api.cpp	176
setup.py	79
benchmark/bench_flash_mla.py	434
benchmark/visualize.py	20
flash_mla/flash_mla_interface.py	34
flash_mla/__init__.py	5