OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
b3861ac8e7
pytorch/torch/_dynamo
History
Xuehai Pan e8fadba28c [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 
The goal of this PR is to provide a standard way to create simple treespec instances and hide the implementation details of the `PyTreeSpec` class.

Changes:

1. Add function `treespec_leaf()` to replace `LeafSpec()`.
2. Add function `treespec_tuple(...)` and `treespec_dict(...)` to create treespec for `tuple` / `dict` which is used for `*args` / `**kwargs`. This avoids direct modification to `treespec` instances that rely on the implementation details of the `PyTreeSpec` class.
3. Change `len(spec.children_specs)` to `spec.num_children`.
4. Change `isinstance(spec, LeafSpec)` to `spec.is_leaf()`.

------

Pull Request resolved: https://github.com/pytorch/pytorch/pull/160843
Approved by: https://github.com/mlazos
2025-11-01 04:12:11 +00:00
..
backends Use 'is' in callable comparisons (#166624) 2025-10-30 19:00:09 +00:00
polyfills [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-11-01 04:12:11 +00:00
repro [devx] Fix invalid symbol definition emitted in fx_graph_runnable.py (#166529) 2025-10-30 16:40:12 +00:00
variables [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-11-01 04:12:11 +00:00
__init__.py [dynamo, nested graph breaks] add disable_nested_graph_breaks decorator/context manager (#166477) 2025-10-31 03:36:27 +00:00
_trace_wrapped_higher_order_op.py [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
aot_compile_types.py Refactor AOTAutogradCacheEntry into AOTAutogradResult (#166656) 2025-10-31 18:54:09 +00:00
aot_compile.py [precompile] Preserve default arguments for dynamo capture (#166654) 2025-11-01 00:12:10 +00:00
bytecode_analysis.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
bytecode_transformation.py [dynamo, 3.14] fix segfault due to improper create_call_function_ex (#166678) 2025-10-31 20:44:53 +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 [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
code_context.py
codegen.py [BE][Typing][Dynamo] Type misc files in torch/_dynamo/variables/ (#166569) 2025-10-31 20:42:27 +00:00
compiled_autograd.py [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
comptime.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
config.py [dynamo] Revert C++-fying of symbolic shape guards (#166427) 2025-10-28 21:57:31 +00:00
convert_frame.py [precompile] Preserve default arguments for dynamo capture (#166654) 2025-11-01 00:12:10 +00:00
create_parameter_op.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
current_scope_id.py
debug_utils.py Use 'is' in callable comparisons (#166624) 2025-10-30 19:00:09 +00:00
decorators.py [dynamo, nested graph breaks] add disable_nested_graph_breaks decorator/context manager (#166477) 2025-10-31 03:36:27 +00:00
device_interface.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
distributed.py Implement guard collectives (optimized version) (#156562) 2025-06-24 04:59:49 +00:00
eval_frame.py [1/N] Remove unused loop variables (#166258) 2025-10-30 12:22:25 +00:00
exc.py [dynamo] clarify graph break handling/logging in symbolic_convert (#166587) 2025-10-31 20:13:16 +00:00
external_utils.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
funcname_cache.py
functional_export.py [export] Update dynamo_graph_capture_for_export to return GraphModule. (#166091) 2025-10-28 04:23:28 +00:00
graph_break_hints.py
graph_break_registry.json [dynamo] clarify graph break handling/logging in symbolic_convert (#166587) 2025-10-31 20:13:16 +00:00
graph_bytecode_inputs.py Revert "[user-streams] Fix stream graph output semantics (#164819)" 2025-10-30 16:53:32 +00:00
graph_deduplication.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
graph_region_tracker.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
graph_utils.py [Intel GPU][pre_compile] Add XPU toolkit version and hardware info in compiled model check. (#162951) 2025-09-18 00:04:22 +00:00
guards.py Revert "[user-streams] Add current stream source (#165211)" 2025-10-30 14:34:43 +00:00
hooks.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
logging.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
metrics_context.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
mutation_guard.py Revert "[dynamo][executorch] Do not trace into exeuctorch LoweredBackendModule (#165126)" 2025-10-10 19:21:41 +00:00
output_graph.py [BE][Typing][Dynamo] Type misc files in torch/_dynamo/variables/ (#166569) 2025-10-31 20:42:27 +00:00
package.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
pgo.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
precompile_context.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
profiler.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
replay_record.py [Dynamo][Better Engineering] Type coverage for torch/_dynamo/utils.py (#159580) 2025-08-04 21:51:53 +00:00
resume_execution.py [dynamo] fix keyerror in resume_execution (again) (#166040) 2025-10-28 07:04:29 +00:00
side_effects.py [BE][Typing][Dynamo] Type misc files in torch/_dynamo/variables/ (#166569) 2025-10-31 20:42:27 +00:00
source.py [BE][Typing][Dynamo] Type misc files in torch/_dynamo/variables/ (#166569) 2025-10-31 20:42:27 +00:00
symbolic_convert.py [dynamo, 3.14] fix segfault due to improper create_call_function_ex (#166678) 2025-10-31 20:44:53 +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 Fix existing Pyrefly errors (#166439) 2025-10-29 02:08:02 +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 Pyrefly suppressions 7/n (#164913) 2025-10-08 07:27:17 +00:00
testing.py Fix error suppression syntax in onnx, jit, _dynamo (#166249) 2025-10-27 02:01:54 +00:00
trace_rules.py [ROCm][CUDA] add unit test utility busy_wait_for_flag (#166218) 2025-10-29 22:40:23 +00:00
types.py [10/N] Apply ruff UP035 rule (#165709) 2025-10-25 00:20:13 +00:00
utils.py Revert "[user-streams] Add current stream source (#165211)" 2025-10-30 14:34:43 +00:00