Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
splitkv_mla.h 162 Bytes
Newer Older
Shengyu Liu's avatar
Reorganize files and add sparse prefill/decoding kernels on hopper  
Shengyu Liu committed
1
2
3
4
5
6
7
8
9
10 
#pragma once

#include "params.h"

namespace sm90 {

template<typename InputT>
void run_flash_splitkv_mla_kernel(DecodingParams &params, cudaStream_t stream);

}