Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
phase1.h 153 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 sm100::fwd::head128 {

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

}