pytorch - 💡(How to fix) Fix DISABLED test_sparse_matmul_cpu_complex64 (__main__.TestSparseCPU)

Traceback (most recent call last): File "/var/lib/jenkins/workspace/test/test_sparse.py", line 3975, in test_sparse_matmul test_error_cases() ~~~~~~~~~~~~~~~~^^ File "/var/lib/jenkins/workspace/test/test_sparse.py", line 3943, in test_error_cases self.assertRaises(RuntimeError, lambda: fn(3, 4, [2, 2, 2], [2, 2, 2])) ~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/testing/_internal/common_utils.py", line 4477, in assertRaises return super().assertRaises(expected_exception, *args, **kwargs) ~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/unittest/case.py", line 813, in assertRaises return context.handle('assertRaises', args, kwargs) ~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/unittest/case.py", line 247, in handle callable_obj(*args, **kwargs) ~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/var/lib/jenkins/workspace/test/test_sparse.py", line 3943, in <lambda> self.assertRaises(RuntimeError, lambda: fn(3, 4, [2, 2, 2], [2, 2, 2])) ~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 2432, 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 2150, 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 735, 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 1919, 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 1535, 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 1594, 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 1438, in compile_frame CompileContext.get().attempt = attempt ~~~~~~~~~~~~~~~~~~^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_guards.py", line 1006, in get raise AssertionError("compile_context is not set") AssertionError: compile_context is not set

To execute this test, run the following from the base repo dir: PYTORCH_TEST_WITH_DYNAMO=1 python test/test_sparse.py TestSparseCPU.test_sparse_matmul_cpu_complex64

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0

TL;DR

The most likely fix for the failing test test_sparse_matmul_cpu_complex64 is to investigate and resolve the AssertionError: compile_context is not set issue in the PyTorch Dynamo compilation process.

Guidance

• Investigate the PyTorch Dynamo compilation process to determine why the compile_context is not being set.
• Review the test case test_sparse_matmul_cpu_complex64 in test_sparse.py to understand the specific conditions that are triggering the error.
• Check the workflow logs and grep for test_sparse_matmul_cpu_complex64 to study the logs and identify any patterns or clues that may help resolve the issue.
• Consider running the test with the PYTORCH_TEST_WITH_DYNAMO=1 environment variable set to 1 and PYTORCH_PRINT_REPRO_ON_FAILURE=1 to gather more information about the failure.

Example

No specific code example is provided as the issue is related to a complex error in the PyTorch Dynamo compilation process.

Notes

The error message suggests that the compile_context is not being set, which may indicate a problem with the PyTorch Dynamo configuration or the test case itself. Further investigation is needed to determine the root cause of the issue.

Recommendation

Apply a workaround by re-enabling the test and running it with the PYTORCH_TEST_WITH_DYNAMO=1 environment variable set to 1 to gather more information about the failure, and then investigate the PyTorch Dynamo compilation process to resolve the AssertionError: compile_context is not set issue.