Skip to content

GitLab

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

#include "params.h"

void run_get_mla_metadata_kernel(GetDecodingMetadataParams &params, cudaStream_t stream);