Joel Schlosser
|
0ff155fb65
|
Fix SDPA for SAM (#115636)
Addresses the regression for Segment Anything Fast in https://github.com/pytorch-labs/segment-anything-fast/issues/99
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115636
Approved by: https://github.com/soulitzer, https://github.com/ani300
|
2023-12-12 18:52:38 +00:00 |
|
soulitzer
|
8885128dcc
|
Fix backward for SDPA NT jagged layout (#115576)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115576
Approved by: https://github.com/jbschlosser, https://github.com/ani300
|
2023-12-12 18:35:40 +00:00 |
|
Antoni Viros
|
1dc4588c6a
|
Add an SDPA dispatcher for nested tensors with jagged layouts (#114164)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/114164
Approved by: https://github.com/jbschlosser
|
2023-12-05 06:33:45 +00:00 |
|
PyTorch MergeBot
|
5cfda9b7f8
|
Revert "Add an SDPA dispatcher for nested tensors with jagged layouts (#114164)"
This reverts commit aafa8233a4.
Reverted https://github.com/pytorch/pytorch/pull/114164 on behalf of https://github.com/malfet due to Broke ROCM, see aafa8233a4 ([comment](https://github.com/pytorch/pytorch/pull/114164#issuecomment-1839798986))
|
2023-12-05 00:35:20 +00:00 |
|
Antoni Viros
|
aafa8233a4
|
Add an SDPA dispatcher for nested tensors with jagged layouts (#114164)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/114164
Approved by: https://github.com/jbschlosser
|
2023-12-04 21:54:02 +00:00 |
|