Add PyTorch version check for FA backend on AMD GPUs#35813

Merged

SunMarc merged 1 commit intohuggingface:mainfrom mht-sharma:sdpa_rocm_version

Jan 22, 2025

+1

Commits on Jan 21, 2025

Disable FA backend for SDPA on AMD GPUs (PyTorch < 2.4.1)
mht-sharma
committed