Skip to content

GitLab

"platforms/reference/include/ReferenceCMAPTorsionIxn.h" did not exist on "bacc1effb7c1e95e133436031cc705cb9c28d380"
Switch branch/tag
Find file Normal viewHistoryPermalink
splitkv_mla.h 149 Bytes
Newer Older
Simon Mo's avatar
Add Sparse Decoding Kernel and Sparse Prefill Kernel for Blackwell  
Simon Mo committed
1
2
3
4
5
6
7
8
9
10 
#pragma once

#include "params.h"

namespace sm100 {

void run_flash_splitkv_mla_fp8_sparse_kernel(DecodingParams &params, cudaStream_t stream);

}