jax icon indicating copy to clipboard operation
jax copied to clipboard
verified publisher icon google

Adding sliding window attention support in cuDNN SDPA API

Open Cjkkkk opened this issue 1 year ago • 2 comments

Cjkkkk avatar Aug 27 '24 22:08 Cjkkkk

@superbobry @kaixih could you guys help take a look?

Cjkkkk avatar Aug 27 '24 22:08 Cjkkkk

LGTM.

kaixih avatar Aug 27 '24 23:08 kaixih

@superbobry all comments addressed, could you take another look? Thanks!

Cjkkkk avatar Aug 29 '24 20:08 Cjkkkk