Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
mla_combine.h 143 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 
#pragma once

#include "params.h"

template<typename ElementT>
void run_flash_mla_combine_kernel(DecodingParams &params, cudaStream_t stream);