Skip to content

[PyTorch] Keep Flex Attention score_mod graphs on the cuDNN backend - #3796

Open
bvolpato wants to merge 1 commit into
NVIDIA:mainfrom
bvolpato:bvolpato/keep-score-mod-graphs-on-cudnn-backend
Open

bvolpato wants to merge 1 commit into
NVIDIA:mainfrom
bvolpato:bvolpato/keep-score-mod-graphs-on-cudnn-backend

[PyTorch] Keep Flex Attention score_mod graphs on the cuDNN backend

78e584f
Select commit
Loading
Failed to load commit list.