mirror of
https://github.com/deepseek-ai/FlashMLA
synced 2025-05-14 16:45:59 +00:00
c2067be3ea
* Fix benchmark script * Performance optimization for compute-bound cases * Add new testcase (s_k = 16384) * Update README.md * Update comment * Update README.md * Add the deep-dive blog * Add background color for MLA Kernel Sched.drawio.svg * Use relative path for the schedule image * Move flash_mla.h to kernels/params.h
6 lines
119 B
C
6 lines
119 B
C
#pragma once
|
|
|
|
#include "params.h"
|
|
|
|
void run_get_mla_metadata_kernel(Mla_metadata_params ¶ms, cudaStream_t stream);
|