pytorch/torch/_dynamo/variables
PyTorch MergeBot 8a233d6000 Revert "[ContextParallel][FlexAttention] Prototype of supporting FlexAttention in Context Parallel (#158692)"
This reverts commit 07fad04181.

Reverted https://github.com/pytorch/pytorch/pull/158692 on behalf of https://github.com/yangw-dev due to failed some internal testapf.metrics.tests.generate_graph_def_test.GenerateGraphDefTest: test_aps_generate_inference_graph_def_with_justknobs1) AssertionError: Expected 'check' to be called once. Called 3 times., please fix the internal test and reland it ([comment](https://github.com/pytorch/pytorch/pull/158692#issuecomment-3140873894))
2025-07-31 18:00:30 +00:00
..
__init__.py [Dynamo] Make frozen dataclasses hashable (#159529) 2025-07-31 07:03:01 +00:00
base.py [dynamo][guards] More small guard optimizations (#159345) 2025-07-29 18:36:49 +00:00
builder.py Various bugfixes for running NanoGPT training (#159166) 2025-07-30 16:30:22 +00:00
builtin.py [iter] Wrap iter(..) call in a ObjectIteratorVariable (#156460) 2025-07-29 21:24:20 +00:00
constant.py Raise exception in Dynamo if op fails in the interpreter (#158661) 2025-07-23 22:31:51 +00:00
ctx_manager.py [BE] fix remaining flake8 v7 warnings (#159044) 2025-07-25 02:56:34 +00:00
dicts.py [Dynamo] Make frozen dataclasses hashable (#159529) 2025-07-31 07:03:01 +00:00
distributed.py [BE][4/16] fix typos in torch/ (torch/_dynamo/) (#156314) 2025-06-23 02:57:19 +00:00
functions.py [dynamo] [guard] Add caching for inside torch.compile.disable function to avoid unnecessary recompilation. (#157566) 2025-07-28 12:44:22 +00:00
higher_order_ops.py Revert "[ContextParallel][FlexAttention] Prototype of supporting FlexAttention in Context Parallel (#158692)" 2025-07-31 18:00:30 +00:00
iter.py [iter] Wrap iter(..) call in a ObjectIteratorVariable (#156460) 2025-07-29 21:24:20 +00:00
lazy.py [dynamo] Avoid recompiling over unused objects (#156891) 2025-07-09 20:14:34 +00:00
lists.py Raise exception in Dynamo if op fails in the interpreter (#158661) 2025-07-23 22:31:51 +00:00
misc.py [Dynamo] [FrozensetSubclass] Add support for user defined frozensets (#154263) 2025-07-04 00:46:05 +00:00
nn_module.py dynamo: handle AttributeErrors from nn_module when infer_paramaters throws. (#158501) 2025-07-30 02:41:41 +00:00
optimizer.py Various bugfixes for running NanoGPT training (#159166) 2025-07-30 16:30:22 +00:00
script_object.py Document dynamo (#146736) 2025-02-13 00:02:21 +00:00
sdpa.py [Dynamo][Misc] Apply typing hints for codegen (#150289) 2025-04-04 14:26:22 +00:00
tensor.py [dynamo][tensor] Skip HASATTR attribute on tensor guards (#158215) 2025-07-15 20:10:47 +00:00
torch_function.py [Bugfix] Fix circular import between export and dynamo from tensor fn map (#158931) 2025-07-24 22:24:57 +00:00
torch.py Graph break with error message (#158800) 2025-07-29 17:34:49 +00:00
user_defined.py [Dynamo] Make frozen dataclasses hashable (#159529) 2025-07-31 07:03:01 +00:00