Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
fwd.h 263 Bytes
Newer Older
shenzhe's avatar
Add DSA MLS sparse prefill dispatch  
shenzhe committed
1
2
3
4
5
6
7
8
9
10
11 
#pragma once

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

namespace gfx93::fwd::dsa_mls {

bool can_run(const SparseAttnFwdParams& params);
bool should_run(const SparseAttnFwdParams& params);
void run(const SparseAttnFwdParams& params);

}  // namespace gfx93::fwd::dsa_mls