pytorch/torch/_dynamo
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
..
backends [ONNX] Remove legacy Dort (#158258) 2025-07-15 19:14:06 +00:00
polyfills [iter] Raise TypeError if iter arg cannot be iterable (#158410) 2025-07-29 21:24:21 +00:00
repro [inductor] fix test_save_graph_repro on Windows. (#159148) 2025-07-25 19:11:08 +00:00
variables Revert "[ContextParallel][FlexAttention] Prototype of supporting FlexAttention in Context Parallel (#158692)" 2025-07-31 18:00:30 +00:00
__init__.py [dynamo] [guard] Add caching for inside torch.compile.disable function to avoid unnecessary recompilation. (#157566) 2025-07-28 12:44:22 +00:00
_trace_wrapped_higher_order_op.py [BE] remove torch deploy - conditionals (#158288) 2025-07-29 17:40:49 +00:00
bytecode_analysis.py [Dynamo][Better Engineering] Add typing to bytecode analysis and transform (#158293) 2025-07-16 21:50:55 +00:00
bytecode_transformation.py [Dynamo][Better Engineering] Add typing to bytecode analysis and transform (#158293) 2025-07-16 21:50:55 +00:00
cache_size.py [Dynamo][Better Engineering] Add typing for comptime, cache, and convert_frame (#158379) 2025-07-18 02:11:57 +00:00
callback.py increment pending_callbacks_counter before initation the pt2 compile callbacks (#157185) 2025-06-30 01:23:59 +00:00
code_context.py
codegen.py [Dynamo][Better Engineering] Support typing in codegen.py (#158386) 2025-07-16 22:09:01 +00:00
compiled_autograd.py Always disable ShardingPropagation cache if compiling (#156868) 2025-07-17 01:33:53 +00:00
comptime.py [Dynamo][Better Engineering] Add typing for comptime, cache, and convert_frame (#158379) 2025-07-18 02:11:57 +00:00
config.py [dynamo] Turn on recursive dict tag optimization (#159186) 2025-07-31 02:36:37 +00:00
convert_frame.py [Precompile] Various small bugfixes, add CachingPrecompile to torchbench (#158847) 2025-07-24 14:09:54 +00:00
create_parameter_op.py
current_scope_id.py
debug_utils.py [inductor] fix test_save_graph_repro on Windows. (#159148) 2025-07-25 19:11:08 +00:00
decorators.py Various bugfixes for running NanoGPT training (#159166) 2025-07-30 16:30:22 +00:00
device_interface.py [Re-land][Inductor] Support native Inductor as backend for MTIA (#159211) 2025-07-29 17:03:24 +00:00
distributed.py Implement guard collectives (optimized version) (#156562) 2025-06-24 04:59:49 +00:00
eval_frame.py [Dynamo][Better Engineering] Add typing annotations to guard and source (#158397) (#159491) 2025-07-30 22:57:50 +00:00
exc.py [dynamo] Added back weblink generation (#159011) 2025-07-24 20:27:11 +00:00
external_utils.py [Dynamo][Better Engineering] Support typing in codegen.py (#158386) 2025-07-16 22:09:01 +00:00
funcname_cache.py gracefully handle tokenize.TokenError in funcname parser. Adds support for non-Python source (#148737) 2025-03-11 19:49:28 +00:00
graph_break_hints.py [dynamo] unimplemented -> unimplemented_v2 in variables/builder.py (#151044) 2025-04-11 09:07:01 +00:00
graph_break_registry.json Graph break with error message (#158800) 2025-07-29 17:34:49 +00:00
graph_deduplication.py [Dynamo] Fix typing in graph_deduplication.py (#152572) 2025-05-13 12:17:59 +00:00
graph_region_tracker.py [BE][4/16] fix typos in torch/ (torch/_dynamo/) (#156314) 2025-06-23 02:57:19 +00:00
graph_utils.py [Hierarchical Compile] Take into account mutation deps in cycle detection (#152506) 2025-05-13 12:17:59 +00:00
guards.py [Dynamo][Better Engineering] Add typing annotations to guard and source (#158397) (#159491) 2025-07-30 22:57:50 +00:00
hooks.py Replace frame_traced_fn hook with get_traced_code() util (#155249) 2025-06-10 22:40:58 +00:00
logging.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
metrics_context.py Track previous MetricsContext edits for ease of debugging. (#159336) 2025-07-29 17:43:52 +00:00
mutation_guard.py Add functionality for installing free variables (#151134) 2025-04-24 17:57:54 +00:00
output_graph.py [Dynamo][Better Engineering] Add typing annotations to guard and source (#158397) (#159491) 2025-07-30 22:57:50 +00:00
package.py Various bugfixes for running NanoGPT training (#159166) 2025-07-30 16:30:22 +00:00
pgo.py Revert "Add torch compile force disable caches alias (#158072)" 2025-07-18 04:58:24 +00:00
precompile_context.py Various bugfixes for running NanoGPT training (#159166) 2025-07-30 16:30:22 +00:00
profiler.py
replay_record.py
resume_execution.py [Dynamo][Better Engineering] Type devices, resume_execution and testing utils (#158593) 2025-07-18 18:22:06 +00:00
side_effects.py [Dynamo][BetterEngineering] Type side_effects.py (#158605) 2025-07-21 19:34:14 +00:00
source.py [Dynamo][Better Engineering] Add typing annotations to guard and source (#158397) (#159491) 2025-07-30 22:57:50 +00:00
symbolic_convert.py Update context in unimplemented_v2 when exception bubbles up to the interpreter (#158924) 2025-07-24 20:50:22 +00:00
tensor_version_op.py [Dynamo][Better Engineering] Type devices, resume_execution and testing utils (#158593) 2025-07-18 18:22:06 +00:00
test_case.py [Dynamo][Better Engineering] Type devices, resume_execution and testing utils (#158593) 2025-07-18 18:22:06 +00:00
test_dont_skip_tracing_functions.py [dynamo] context manager/decorator for dynamo config patching during tracing (#150586) 2025-04-23 09:12:13 +00:00
test_minifier_common.py [inductor] normalize path of the code. (#159255) 2025-07-28 23:42:11 +00:00
testing.py [precompile] Add CompilePackage to serialize dynamo states. (#155118) 2025-06-13 13:54:10 +00:00
trace_rules.py [BE] Make PyObjectSlot use a global PyInterpreter and remove (#158427) 2025-07-30 17:29:43 +00:00
types.py Implement guard collectives (optimized version) (#156562) 2025-06-24 04:59:49 +00:00
utils.py [Dynamo][Better Engineering] Add typing annotations to guard and source (#158397) (#159491) 2025-07-30 22:57:50 +00:00