pytorch - 💡(How to fix) Fix DISABLED test_static_input_indices (__main__.TestAOTJointWithDescriptors) [1 comments, 1 participants]

Traceback (most recent call last): File "/__w/pytorch/pytorch/test/functorch/test_aot_joint_with_descriptors.py", line 1073, in test_static_input_indices class SimpleLinear(nn.Module): File "/__w/pytorch/pytorch/test/functorch/test_aot_joint_with_descriptors.py", line 1081, in torch_dynamo_resume_in_test_static_input_indices_at_1073 model = SimpleLinear() File "/__w/pytorch/pytorch/test/functorch/test_aot_joint_with_descriptors.py", line 1083, in torch_dynamo_resume_in_test_static_input_indices_at_1081 gm = dynamo_graph_capture_for_export(model)(*inputs) File "/__w/pytorch/pytorch/test/functorch/test_aot_joint_with_descriptors.py", line 1083, in torch_dynamo_resume_in_test_static_input_indices_at_1083 gm = dynamo_graph_capture_for_export(model)(*inputs) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/functional_export.py", line 809, in inner out = fullgraph_capture( fn, ...<2 lines>... constraints=constraints, ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 1344, in fullgraph_capture return _fullgraph_capture_frame( frame, constraints=constraints, _is_export_deprecated_do_not_use=_is_export_deprecated_do_not_use, ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 1413, in _fullgraph_capture_frame dynamo_output.graph_capture_output(frame.argdefs, frame.kwdefaults), ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 991, in graph_capture_output return GraphCaptureOutput( OutputGraphCommon( ...<12 lines>... self.tracer_output.f_globals, ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 2480, in call result = self._torchdynamo_orig_backend( frame, cache_entry, self.hooks, frame_state, skip=1 ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 2198, in call result = self._inner_convert( frame, cache_entry, hooks, frame_state, skip=skip + 1 ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 758, in call result = _compile( frame.f_code, ...<16 lines>... convert_frame_box=self._box, ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 2048, in _compile raise InternalTorchDynamoError( f"{type(e).qualname}: {str(e)}" ).with_traceback(e.traceback) from None File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 1983, in _compile guarded_code, tracer_output = compile_inner(code, one_graph, hooks) ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_utils_internal.py", line 96, in wrapper_function return function(*args, **kwargs) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 1593, in compile_inner result = _compile_inner(code, one_graph, hooks) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 1652, in _compile_inner dynamo_output = compile_frame( code, ...<11 lines>... package=package, ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 1500, in compile_frame bytecode, tracer_output = transform_code_object(code, transform) ~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/bytecode_transformation.py", line 1626, in transform_code_object tracer_output = transformations(instructions, code_options) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 1472, in transform tracer_output = trace_frame( code, ...<14 lines>... package=package, ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 365, in _fn return fn(*args, **kwargs) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 933, in trace_frame run_tracer() ~~~~~~~~~~^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 914, in run_tracer tracer.run() ~~~~~~~~~~^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/symbolic_convert.py", line 1839, in run while self.step(): ~~~~~~~~~^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/symbolic_convert.py", line 1506, in step self.dispatch_table[inst.opcode](self, inst) ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/symbolic_convert.py", line 1043, in wrapper return inner_fn(self, inst) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/symbolic_convert.py", line 2974, in STORE_ATTR VariableTracker.build(self, setattr).call_function( ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^ self, # type: ignore[arg-type] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [obj, VariableTracker.build(self, inst.argval), val], ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ {}, ^^^ ) ^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/variables/builtin.py", line 1519, in call_function return handler(tx, args, kwargs) # type: ignore[return-value] File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/variables/builtin.py", line 1078, in <lambda> tx, [v.realize() for v in args], kwargs ~~~~~~~~~^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/variables/lazy.py", line 107, in realize self._cache.realize() ~~~~~~~~~~~~~~~~~~~^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/variables/lazy.py", line 44, in realize self.vt = builder.VariableBuilder( ~~~~~~~~~~~~~~~~~~~~~~~~ tx, self.source, allow_lazy_constant=False ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ )(self.value) ~^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/variables/builder.py", line 488, in call return self._call_impl(value) ~~~~~~~~~~~~~~~^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/variables/builder.py", line 525, in _call_impl vt = self._wrap(value) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/variables/builder.py", line 878, in _wrap result = dict( build_key_value(i, k, v) for i, k, v in enumerate_items_with_dict_position(value) ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/variables/builder.py", line 880, in <genexpr> for i, k, v in enumerate_items_with_dict_position(value) ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/utils.py", line 2944, in <genexpr> return ((i, k, v) for i, (k, v) in enumerate(items)) ~~~~~~~~~^^^^^^^ torch._dynamo.exc.InternalTorchDynamoError: RuntimeError: dictionary changed size during iteration

from user code: File "<string>", line 10, in init

Set TORCHDYNAMO_VERBOSE=1 for the internal stack trace (please do this especially if you're reporting a bug to PyTorch). For even more developer context, set TORCH_LOGS="+dynamo"

To execute this test, run the following from the base repo dir: PYTORCH_TEST_WITH_DYNAMO=1 python test/functorch/test_aot_joint_with_descriptors.py TestAOTJointWithDescriptors.test_static_input_indices

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0

TL;DR

The test test_static_input_indices is failing due to an InternalTorchDynamoError caused by a RuntimeError: dictionary changed size during iteration, and the fix likely involves modifying the test or the dynamo configuration to avoid this error.

Guidance

• Review the test test_static_input_indices in functorch/test_aot_joint_with_descriptors.py to identify potential issues with dictionary iteration.
• Set TORCHDYNAMO_VERBOSE=1 and TORCH_LOGS="+dynamo" to obtain more detailed logs and internal stack traces for debugging.
• Run the test with PYTORCH_TEST_WITH_DYNAMO=1 python test/functorch/test_aot_joint_with_descriptors.py TestAOTJointWithDescriptors.test_static_input_indices to reproduce the error.
• Investigate the dynamo configuration and the test's usage of dictionaries to determine the cause of the dictionary changed size during iteration error.

Notes

The error message suggests a problem with dictionary iteration, but the root cause is unclear without further investigation. The provided logs and stack traces may help identify the issue.

Recommendation

Apply a workaround by modifying the test or the dynamo configuration to avoid the dictionary changed size during iteration error, as the exact fix is unclear without further debugging.