torch._dynamo.exc.TorchRuntimeError: RuntimeError when making fake tensor call Explanation: Dynamo failed to run FX node with fake tensors: call_method sin(*(BatchedTensor(lvl=1, bdim=0, value= FakeTensor(..., size=(3,)) ),), **{}): got RuntimeError('tls_on_entry.has_value() INTERNAL ASSERT FAILED at "/__w/pytorch/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp":53, please report a bug to PyTorch. ') Hint: Your code may result in an error when running in eager. Please double check that your code doesn't contain a similar error when actually running eager/uncompiled. You can do this by removing the torch.compile call, or by using torch.compiler.set_stance("force_eager").

Developer debug context:

For more details about this graph break, please visit: https://meta-pytorch.github.io/compile-graph-break-site/gb/gb4315.html

from user code: File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_functorch/apis.py", line 220, in wrapped return vmap_impl( File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_functorch/vmap.py", line 316, in vmap_impl return _flat_vmap( File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_functorch/vmap.py", line 507, in _flat_vmap batched_outputs = func(*batched_inputs, **kwargs) File "/var/lib/jenkins/workspace/test/functorch/test_eager_transforms.py", line 3784, in forward return x.sin()

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"

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/var/lib/jenkins/workspace/test/functorch/test_eager_transforms.py", line 3780, in test_autograd_function_no_setup_context class MySin(torch.autograd.Function): File "/var/lib/jenkins/workspace/test/functorch/test_eager_transforms.py", line 3793, in torch_dynamo_resume_in_test_autograd_function_no_setup_context_at_3780 with self.assertRaisesRegex(RuntimeError, "must override the setup_context"): File "/var/lib/jenkins/workspace/test/functorch/test_eager_transforms.py", line 3793, in torch_dynamo_resume_in_test_autograd_function_no_setup_context_at_3793 with self.assertRaisesRegex(RuntimeError, "must override the setup_context"): ~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/unittest/case.py", line 285, in exit self._raiseFailure('"{}" does not match "{}"'.format( ~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ expected_regex.pattern, str(exc_value))) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/unittest/case.py", line 209, in _raiseFailure raise self.test_case.failureException(msg) AssertionError: "must override the setup_context" does not match "RuntimeError when making fake tensor call Explanation: Dynamo failed to run FX node with fake tensors: call_method sin(*(BatchedTensor(lvl=1, bdim=0, value= FakeTensor(..., size=(3,)) ),), **{}): got RuntimeError('tls_on_entry.has_value() INTERNAL ASSERT FAILED at "/__w/pytorch/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp":53, please report a bug to PyTorch. ') Hint: Your code may result in an error when running in eager. Please double check that your code doesn't contain a similar error when actually running eager/uncompiled. You can do this by removing the torch.compile call, or by using torch.compiler.set_stance("force_eager").

Developer debug context:

For more details about this graph break, please visit: https://meta-pytorch.github.io/compile-graph-break-site/gb/gb4315.html

from user code: File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_functorch/apis.py", line 220, in wrapped return vmap_impl( File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_functorch/vmap.py", line 316, in vmap_impl return _flat_vmap( File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_functorch/vmap.py", line 507, in _flat_vmap batched_outputs = func(*batched_inputs, **kwargs) File "/var/lib/jenkins/workspace/test/functorch/test_eager_transforms.py", line 3784, in forward return x.sin()

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_eager_transforms.py TestComposabilityCPU.test_autograd_function_no_setup_context_transform_vmap_cpu

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0

TL;DR

The most likely fix for the failing test is to modify the test to handle the error when running in eager mode or to remove the torch.compile call.

Guidance

• The error message suggests that the test is failing due to an internal assert failure in PyTorch, specifically when running with fake tensors.
• To debug this issue, try removing the torch.compile call or using torch.compiler.set_stance("force_eager") to run the test in eager mode.
• The test can be executed using the command PYTORCH_TEST_WITH_DYNAMO=1 python test/functorch/test_eager_transforms.py TestComposabilityCPU.test_autograd_function_no_setup_context_transform_vmap_cpu.
• Setting TORCHDYNAMO_VERBOSE=1 can provide more detailed internal stack traces, and setting TORCH_LOGS="+dynamo" can provide additional developer context.

Example

No specific code snippet is provided, but the error message suggests modifying the forward method in test_eager_transforms.py to handle the error when running in eager mode.

Notes

The issue seems to be related to the interaction between PyTorch's Dynamo and the vmap functionality. The error message provides a hint to check the code for similar errors when running in eager mode.

Recommendation

Apply a workaround by modifying the test to handle the error when running in eager mode or removing the torch.compile call, as the root cause of the issue is not immediately clear and may require further investigation.