OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
afa1eda901
pytorch/torch/_dynamo/variables
History
Eddie Yan 0dcd482e54 [SDPA] Respect sdpa_kernel's priority_order setting in torch.compile (#147768) 
[https://github.com/pytorch/pytorch/pull/140467](https://github.com/pytorch/pytorch/pull/140467) added the option to specify a priority order for SDPA but the `torch.compile` path silently ignored this setting as I wasn't aware of the separate context manager handling on `torch.compile`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/147768
Approved by: https://github.com/drisspg
2025-03-13 18:52:34 +00:00
..
__init__.py Introduce UserDefinedExceptionClassVariable (#146504) 2025-03-11 18:55:45 +00:00
base.py [Dynamo] Replace unimplemented withunimplemented_v2 in torch/_dynamo/variables/base.py (#148177) 2025-03-13 06:35:51 +00:00
builder.py Introduce UserDefinedExceptionClassVariable (#146504) 2025-03-11 18:55:45 +00:00
builtin.py Add __context/cause/suppress_context/traceback__ to Exception (#146499) 2025-03-11 18:55:45 +00:00
constant.py Replace unimplemented with unimplemented_v2 for dynamo (#148158) 2025-03-03 21:00:17 +00:00
ctx_manager.py [SDPA] Respect sdpa_kernel's priority_order setting in torch.compile (#147768) 2025-03-13 18:52:34 +00:00
dicts.py [BE][PYFMT] migrate PYFMT for torch._dynamo to ruff format (#144549) 2025-02-28 03:03:53 +00:00
distributed.py [Dyamo] Replace unimplemented with unimplemented_v2 for variables/distributed (#148500) 2025-03-05 20:41:43 +00:00
functions.py Correctly propagate exception to parent tx (#146502) 2025-03-11 18:55:45 +00:00
higher_order_ops.py [scan] Flattened output of HOP scan (#148955) 2025-03-12 18:27:27 +00:00
iter.py Document dynamo (#146736) 2025-02-13 00:02:21 +00:00
lazy.py PEP585 update - torch/_dynamo (#145105) 2025-01-18 20:47:11 +00:00
lists.py [dynamo] allow global import from collections import deque in user code (#148676) 2025-03-10 13:14:05 +00:00
misc.py Add __context/cause/suppress_context/traceback__ to Exception (#146499) 2025-03-11 18:55:45 +00:00
nn_module.py [dynamo] more better error messages [3/N] (#147494) 2025-02-28 06:23:28 +00:00
optimizer.py [dynamo][optimizers] Install ID_GUARDED tensors into the Fx graph (#147824) 2025-02-28 03:22:11 +00:00
script_object.py Document dynamo (#146736) 2025-02-13 00:02:21 +00:00
sdpa.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
tensor.py [BE][PYFMT] migrate PYFMT for torch._dynamo to ruff format (#144549) 2025-02-28 03:03:53 +00:00
torch_function.py [BE][PYFMT] migrate PYFMT for torch._dynamo to ruff format (#144549) 2025-02-28 03:03:53 +00:00
torch.py [SDPA] Respect sdpa_kernel's priority_order setting in torch.compile (#147768) 2025-03-13 18:52:34 +00:00
user_defined.py Add __context/cause/suppress_context/traceback__ to Exception (#146499) 2025-03-11 18:55:45 +00:00