FlashMLA/csrc/kernels/get_mla_metadata.h
2025-04-21 17:22:59 +08:00

6 lines
122 B
C

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