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 ¶ms, cudaStream_t stream);