Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
phase1.h 175 Bytes
Newer Older
Shengyu Liu's avatar
Multiple updates and refactorings (#150)  
Shengyu Liu committed
1
2
3
4
5
6
7
8
9
10 
#pragma once

#include "../../../params.h"

namespace sm90::fwd {

template<int D_QK, bool HAVE_TOPK_LENGTH>
void run_fwd_phase1_kernel(const SparseAttnFwdParams& params);

}