Add SDPA support for PatchTST model #42465
kashif
approved these changes
on 2025-11-28
vasqu
commented
on 2025-11-28
Add SDPA and Flash Attention support for PatchTST model
23fe9ff8
Guard PatchTST positional init under ZeRO-3
1869b172
Force SDPA in PatchTST regression integration test
d01e82d6
Use sdpa attn in PatchTST regression test
c53a0cf4
Merge branch 'main' into add-sdpa-support-patchtst
eaea5797
fixups re tests
3c6e82e1
Merge branch 'main' into add-sdpa-support-patchtst
b05f61b9
vasqu
approved these changes
on 2025-12-01
vasqu
enabled auto-merge (squash) 309 days ago
vasqu
merged
dd6cfdd3
into main 309 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub