pytorch/torch/_dynamo
Edward Z. Yang cc8f1cddd4 Turn on type-checking in torch.fx.experimental.symbolic_shapes (#136972)
Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/136972
Approved by: https://github.com/Skylion007
ghstack dependencies: #136934, #136935
2024-10-01 13:22:10 +00:00
..
backends [Dynamo] Use custom backend to reenter metadata tf mode when tracing while/cond (#134732) 2024-09-14 18:52:22 +00:00
polyfills Revert "[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)" (#136590) 2024-09-25 21:10:43 +00:00
repro [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00
variables Allow automatic dynamic shapes for closures and set current node properly in flexattention subgraph lowering (#137043) 2024-10-01 09:08:08 +00:00
__init__.py Revert "[dynamo] Cache _dynamo.disable results (#134272)" 2024-08-28 21:51:43 +00:00
_trace_wrapped_higher_order_op.py [hop] require hops to override __call__. (#134352) 2024-08-28 19:56:40 +00:00
bytecode_analysis.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
bytecode_transformation.py [dynamo] refactor resume_execution.py to use bytecode templates (#136483) 2024-09-24 22:20:26 +00:00
cache_size.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
callback.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
code_context.py add types to _dynamo/code_context.py (#136665) 2024-09-27 18:27:42 +00:00
codegen.py Correctly convert Python float to float64 when passing argument as Tensor (#136413) 2024-09-26 16:50:13 +00:00
compiled_autograd.py [compiled autograd] undo view_to_reshape inductor fx pass in node name matching (#136741) 2024-10-01 03:22:49 +00:00
comptime.py Improve print stack/locals printing in comptime (#133651) 2024-08-27 01:29:50 +00:00
config.py [Dynamo] Mark config.dead_code_elimination as deprecated (#136933) 2024-10-01 03:51:59 +00:00
convert_frame.py [Dynamo] Mark config.dead_code_elimination as deprecated (#136933) 2024-10-01 03:51:59 +00:00
create_parameter_op.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
current_scope_id.py add typing to _dynamo/current_scope_id.py (#136676) 2024-09-27 04:09:15 +00:00
debug_utils.py [debug] Add helper to run cProfile on a function (#135084) 2024-09-05 23:41:30 +00:00
decorators.py [dynamo] simplify implementation for functools.reduce (#133778) 2024-09-16 04:53:06 +00:00
device_interface.py Add Triton CPU as an Inductor backend (#133408) 2024-09-30 20:24:52 +00:00
distributed.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
eval_frame.py dynamo: Only log if we've disabled eval_frame once. (#134529) 2024-08-30 00:35:25 +00:00
exc.py [dynamo] attempt run only mode when dynamo cache limit is hit (#136655) 2024-09-27 17:15:05 +00:00
external_utils.py [compiled autograd] match eager behavior for ctx.saved_variables (#134286) 2024-08-24 12:06:36 +00:00
funcname_cache.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
guards.py Propagate detailed location information of shape guards to guards/recompiles output (#136917) 2024-09-30 00:43:12 +00:00
hooks.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
logging.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
mutation_guard.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
output_graph.py Turn on type-checking in torch.fx.experimental.symbolic_shapes (#136972) 2024-10-01 13:22:10 +00:00
profiler.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
replay_record.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
resume_execution.py Revert "[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)" (#136590) 2024-09-25 21:10:43 +00:00
side_effects.py Revert "[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)" (#136590) 2024-09-25 21:10:43 +00:00
source.py Allow automatic dynamic shapes for closures and set current node properly in flexattention subgraph lowering (#137043) 2024-10-01 09:08:08 +00:00
symbolic_convert.py Revert "[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)" (#136590) 2024-09-25 21:10:43 +00:00
tensor_version_op.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
test_case.py [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
test_minifier_common.py [inductor] fix _maybe_subprocess_run not support Windows path (#134365) 2024-08-26 17:48:11 +00:00
testing.py Revert "[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)" (#136590) 2024-09-25 21:10:43 +00:00
trace_rules.py Revert "[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)" (#136590) 2024-09-25 21:10:43 +00:00
types.py Fix PT2 Source Code Annotations (#136460) 2024-09-28 03:54:43 +00:00
utils.py Revert "[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)" (#136590) 2024-09-25 21:10:43 +00:00