transformers
Fix SDPA correctness following torch==2.1.2 regression
#27973

Merged

Fix SDPA correctness following torch==2.1.2 regression #27973

fxmarty merged 4 commits into huggingface:main from fxmarty:fix-sdpa-torch-212

fix sdpa with non-contiguous inputs for gpt_bigcode

3ac1d30e

fix other archs

8c6044fa

fxmarty requested a review from

amyeroberts 2 years ago

fxmarty requested a review from

ArthurZucker 2 years ago

fxmarty requested a review from

LysandreJik 2 years ago

add currently comment

fc076dfe

format

657abfa7

amyeroberts approved these changes on 2023-12-12

fxmarty merged 78172dcd into main 2 years ago

Reviewers

amyeroberts

ArthurZucker

LysandreJik

Assignees

No one assigned

Labels

None yet

Milestone

No milestone