Skip to content

Add TK Wave kernels to attention benchmark for attention-v transpose - #44

Open
raikonenfnu wants to merge 2 commits into
nod-ai:mainfrom
raikonenfnu:waveAttn
Open

Add TK Wave kernels to attention benchmark for attention-v transpose#44
raikonenfnu wants to merge 2 commits into
nod-ai:mainfrom
raikonenfnu:waveAttn

Conversation

@raikonenfnu

@raikonenfnu raikonenfnu commented Jan 29, 2025

Copy link
Copy Markdown
Member
  • Modify problem to be transposeV attention by default
  • Add Wave helper functions for attention transposeV

Signed-off-by: Stanley Winata <stanley.winata@amd.com>
@raikonenfnu
raikonenfnu requested a review from saienduri January 29, 2025 16:22
Signed-off-by: Stanley Winata <stanley.winata@amd.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant