csrc/kernels/get_mla_metadata.h (3 lines of code) (raw):

#pragma once #include "params.h" void run_get_mla_metadata_kernel(Mla_metadata_params &params, cudaStream_t stream);