Michael Lazos 27dee935af [Dynamo] Ensure torch function modes are dispatched on builtin ops (#137117) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/137117 Approved by: https://github.com/yanboliang, https://github.com/williamwen42 ghstack dependencies: #137114, #137115, #137116		2024-10-09 02:29:40 +00:00
..
__init__.py	[dynamo] support torch.nn.attention.sdpa_kernel context manager (#135404)	2024-09-12 22:04:48 +00:00
base.py	Revert "Fix constant propagation in builtins and UserClasses (#131354)"	2024-09-25 23:01:03 +00:00
builder.py	[Dynamo] Trace enter/exit of TorchFunctionModes (#135422) (#137114)	2024-10-09 02:29:40 +00:00
builtin.py	[Dynamo] Ensure torch function modes are dispatched on builtin ops (#137117)	2024-10-09 02:29:40 +00:00
constant.py	[dynamo] Fix support for classmethod(property(...)) (#134968)	2024-09-18 04:47:51 +00:00
ctx_manager.py	[Dynamo] Trace enter/exit of TorchFunctionModes (#135422) (#137114)	2024-10-09 02:29:40 +00:00
dicts.py	Remove vt argument in raise_observed_exception (#136037)	2024-09-24 02:36:57 +00:00
distributed.py	Revert "Fix constant propagation in builtins and UserClasses (#131354)"	2024-09-25 23:01:03 +00:00
functions.py	Error message for allow_in_graph decorator and arbitrary function combo (#135972)	2024-10-08 22:48:38 +00:00
higher_order_ops.py	Revert "[Dynamo] Move flex attention torch function mode to traceable HOP file (#137120)"	2024-10-08 17:26:19 +00:00
iter.py	Remove vt argument in raise_observed_exception (#136037)	2024-09-24 02:36:57 +00:00
lazy.py	Revert "[Dynamo] Handle torch function subclass/mode dispatch on generic tensor methods (#137119)"	2024-10-08 17:09:26 +00:00
lists.py	[torch.compile] torch.vmap supports dynamic shapes + enable flex attention create_block_mask dynamic shapes (#137163)	2024-10-04 05:16:04 +00:00
misc.py	Revert "[Dynamo] Move flex attention torch function mode to traceable HOP file (#137120)"	2024-10-08 17:26:19 +00:00
nn_module.py	Remove vt argument in raise_observed_exception (#136037)	2024-09-24 02:36:57 +00:00
optimizer.py	[aotd] Cleanup on subclasses in inductor freezing (#136549)	2024-09-27 11:20:03 +00:00
script_object.py	Add basic mypy annotations to dynamo (#132415)	2024-08-04 18:43:36 +00:00
sdpa.py	Grouped Query Attention (#132689)	2024-08-07 05:35:36 +00:00
tensor.py	[Dynamo] Ensure torch function modes are dispatched on builtin ops (#137117)	2024-10-09 02:29:40 +00:00
torch_function.py	[Dynamo] Ensure torch function modes are dispatched on builtin ops (#137117)	2024-10-09 02:29:40 +00:00
torch.py	[Dynamo] Ensure torch function modes are dispatched on builtin ops (#137117)	2024-10-09 02:29:40 +00:00
user_defined.py	[Dynamo] Trace enter/exit of TorchFunctionModes (#135422) (#137114)	2024-10-09 02:29:40 +00:00