Issue Station
Sign In
Tools hereToolbox

pytorch - 💡(How to fix) Fix DISABLED test_can_use_functionalize_when_key_is_excluded_cpu (__main__.TestComposabilityCPU)

Official PRs (…)
ON THIS PAGE

Recommended Tools

×6

Utilities matched from this issue’s tags and category — try them while you read without losing context.

GitHub issue graph ai analysis

Paste a GitHub issue URL. We fetch that issue, discover linked issues from bodies/comments/timeline, collect linked pull requests, and produce a structured English report.

The report is written in English Markdown for sharing and archival.

Helpful · Quick feedback

Loading…

Error Message

Traceback (most recent call last): File "/__w/pytorch/pytorch/test/functorch/test_eager_transforms.py", line 3873, in test_can_use_functionalize_when_key_is_excluded self.assertEqual(gm(x), expected) ~~^^^ File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 1000, in call_wrapped return self._wrapped_call(self, *args, **kwargs) ~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 507, in call raise e File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 493, in call return super(self.cls, obj).call(*args, **kwargs) # type: ignore[misc] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1778, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1789, in _call_impl return forward_call(*args, **kwargs) File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 2620, 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.14/site-packages/torch/_dynamo/convert_frame.py", line 2311, in call result = self._inner_convert( frame, cache_entry, hooks, frame_state, skip=skip + 1 ) File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 777, in call result = _compile( frame.f_code, ...<18 lines>... convert_frame_box=self._box, ) File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 2094, in _compile guarded_code, tracer_output = compile_inner(code, one_graph, hooks) ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_utils_internal.py", line 96, in wrapper_function return function(*args, **kwargs) File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 1679, in compile_inner result = _compile_inner(code, one_graph, hooks) File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 1895, in _compile_inner check_fn = dynamo_output.build_guards( code, ...<2 lines>... cache_entries=cache_entries, ) File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 999, in build_guards return CheckFunctionManager( code, ...<5 lines>... strict_error=strict_error, ) File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/guards.py", line 4501, in init raise AssertionError( ...<2 lines>... ) AssertionError: Guard failed on the same frame it was created. This is a bug - please create an issue.Guard fail reason: 3/0: tensor 'x_1' dispatch key set mismatch. expected DispatchKeySet(CPU, BackendSelect, ADInplaceOrView, AutogradCPU), actual DispatchKeySet(CPU, BackendSelect)

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_can_use_functionalize_when_key_is_excluded_cpu

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0

Root Cause

This test was disabled because it is failing in CI. See recent examples and the most recent trunk workflow logs.

Fix Action

Fix / Workaround

Traceback (most recent call last):
  File "/__w/pytorch/pytorch/test/functorch/test_eager_transforms.py", line 3873, in test_can_use_functionalize_when_key_is_excluded
    self.assertEqual(gm(x), expected)
                     ~~^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 1000, in call_wrapped
    return self._wrapped_call(self, *args, **kwargs)
           ~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 507, in __call__
    raise e
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 493, in __call__
    return super(self.cls, obj).__call__(*args, **kwargs)  # type: ignore[misc]
           ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1778, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1789, in _call_impl
    return forward_call(*args, **kwargs)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 2620, 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.14/site-packages/torch/_dynamo/convert_frame.py", line 2311, in __call__
    result = self._inner_convert(
        frame, cache_entry, hooks, frame_state, skip=skip + 1
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 777, in __call__
    result = _compile(
        frame.f_code,
    ...<18 lines>...
        convert_frame_box=self._box,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 2094, in _compile
    guarded_code, tracer_output = compile_inner(code, one_graph, hooks)
                                  ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_utils_internal.py", line 96, in wrapper_function
    return function(*args, **kwargs)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 1679, in compile_inner
    result = _compile_inner(code, one_graph, hooks)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 1895, in _compile_inner
    check_fn = dynamo_output.build_guards(
        code,
    ...<2 lines>...
        cache_entries=cache_entries,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 999, in build_guards
    return CheckFunctionManager(
        code,
    ...<5 lines>...
        strict_error=strict_error,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/guards.py", line 4501, in __init__
    raise AssertionError(
    ...<2 lines>...
    )
AssertionError: Guard failed on the same frame it was created. This is a bug - please create an issue.Guard fail reason: 3/0: tensor 'x_1' dispatch key set mismatch. expected DispatchKeySet(CPU, BackendSelect, ADInplaceOrView, AutogradCPU), actual DispatchKeySet(CPU, BackendSelect)

Code Example

Traceback (most recent call last):
  File "/__w/pytorch/pytorch/test/functorch/test_eager_transforms.py", line 3873, in test_can_use_functionalize_when_key_is_excluded
    self.assertEqual(gm(x), expected)
                     ~~^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 1000, in call_wrapped
    return self._wrapped_call(self, *args, **kwargs)
           ~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 507, in __call__
    raise e
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 493, in __call__
    return super(self.cls, obj).__call__(*args, **kwargs)  # type: ignore[misc]
           ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1778, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1789, in _call_impl
    return forward_call(*args, **kwargs)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 2620, 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.14/site-packages/torch/_dynamo/convert_frame.py", line 2311, in __call__
    result = self._inner_convert(
        frame, cache_entry, hooks, frame_state, skip=skip + 1
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 777, in __call__
    result = _compile(
        frame.f_code,
    ...<18 lines>...
        convert_frame_box=self._box,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 2094, in _compile
    guarded_code, tracer_output = compile_inner(code, one_graph, hooks)
                                  ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_utils_internal.py", line 96, in wrapper_function
    return function(*args, **kwargs)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 1679, in compile_inner
    result = _compile_inner(code, one_graph, hooks)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 1895, in _compile_inner
    check_fn = dynamo_output.build_guards(
        code,
    ...<2 lines>...
        cache_entries=cache_entries,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 999, in build_guards
    return CheckFunctionManager(
        code,
    ...<5 lines>...
        strict_error=strict_error,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/guards.py", line 4501, in __init__
    raise AssertionError(
    ...<2 lines>...
    )
AssertionError: Guard failed on the same frame it was created. This is a bug - please create an issue.Guard fail reason: 3/0: tensor 'x_1' dispatch key set mismatch. expected DispatchKeySet(CPU, BackendSelect, ADInplaceOrView, AutogradCPU), actual DispatchKeySet(CPU, BackendSelect)


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_can_use_functionalize_when_key_is_excluded_cpu

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0
RAW_BUFFERClick to expand / collapse

Platforms: dynamo

This test was disabled because it is failing in CI. See recent examples and the most recent trunk workflow logs.

Over the past 6 hours, it has been determined flaky in 8 workflow(s) with 8 failures and 8 successes.

Debugging instructions (after clicking on the recent samples link): DO NOT ASSUME THINGS ARE OKAY IF THE CI IS GREEN. We now shield flaky tests from developers so CI will thus be green but it will be harder to parse the logs. To find relevant log snippets:

  1. Click on the workflow logs linked above
  2. Click on the Test step of the job so that it is expanded. Otherwise, the grepping will not work.
  3. Grep for test_can_use_functionalize_when_key_is_excluded_cpu
  4. There should be several instances run (as flaky tests are rerun in CI) from which you can study the logs.
<details><summary>Sample error message</summary> 
Traceback (most recent call last):
  File "/__w/pytorch/pytorch/test/functorch/test_eager_transforms.py", line 3873, in test_can_use_functionalize_when_key_is_excluded
    self.assertEqual(gm(x), expected)
                     ~~^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 1000, in call_wrapped
    return self._wrapped_call(self, *args, **kwargs)
           ~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 507, in __call__
    raise e
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/fx/graph_module.py", line 493, in __call__
    return super(self.cls, obj).__call__(*args, **kwargs)  # type: ignore[misc]
           ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1778, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1789, in _call_impl
    return forward_call(*args, **kwargs)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 2620, 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.14/site-packages/torch/_dynamo/convert_frame.py", line 2311, in __call__
    result = self._inner_convert(
        frame, cache_entry, hooks, frame_state, skip=skip + 1
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 777, in __call__
    result = _compile(
        frame.f_code,
    ...<18 lines>...
        convert_frame_box=self._box,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 2094, in _compile
    guarded_code, tracer_output = compile_inner(code, one_graph, hooks)
                                  ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_utils_internal.py", line 96, in wrapper_function
    return function(*args, **kwargs)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 1679, in compile_inner
    result = _compile_inner(code, one_graph, hooks)
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 1895, in _compile_inner
    check_fn = dynamo_output.build_guards(
        code,
    ...<2 lines>...
        cache_entries=cache_entries,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/convert_frame.py", line 999, in build_guards
    return CheckFunctionManager(
        code,
    ...<5 lines>...
        strict_error=strict_error,
    )
  File "/opt/conda/envs/py_3.14/lib/python3.14/site-packages/torch/_dynamo/guards.py", line 4501, in __init__
    raise AssertionError(
    ...<2 lines>...
    )
AssertionError: Guard failed on the same frame it was created. This is a bug - please create an issue.Guard fail reason: 3/0: tensor 'x_1' dispatch key set mismatch. expected DispatchKeySet(CPU, BackendSelect, ADInplaceOrView, AutogradCPU), actual DispatchKeySet(CPU, BackendSelect)


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_can_use_functionalize_when_key_is_excluded_cpu

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0
</details>

Test file path: functorch/test_eager_transforms.py

For all disabled tests (by GitHub issue), see https://hud.pytorch.org/disabled.

cc @chauhang @penguinwu @Chillee @samdow @kshitij12345

Vote matrix · Quick signals

Works
Did the solution work? Tap to confirm.
Easy Fix
Was it a quick fix?
Time Saver
Did it save you time?
Blocking
Was it severely blocking?
Common Issue
Are others likely hitting this too?
Flaky / Intermittent
Is it intermittent?
Verified / Reproducible
Can you reproduce it reliably?
Loading…

Still need to ship something?

×6

Another batch ranked right after the header list — different links, same matching logic.

Back to top recommendations

TRENDING

pytorch - 💡(How to fix) Fix DISABLED test_can_use_functionalize_when_key_is_excluded_cpu (__main__.TestComposabilityCPU)