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

Traceback (most recent call last): File "/var/lib/jenkins/workspace/test/test_sparse.py", line 1780, in test_sparse_addmm test_shape(7, 8, 9, 20, False, None) File "/var/lib/jenkins/workspace/test/test_sparse.py", line 1781, in torch_dynamo_resume_in_test_sparse_addmm_at_1780 test_shape(7, 8, 9, 20, True, None) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 2474, 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 2192, 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 736, 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 1961, 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 1571, 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 1630, 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 1474, 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_addmm_cpu_float16

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_addmm_cpu_float16 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 file test_sparse.py and the PyTorch Dynamo code to understand the expected behavior and potential causes of the error.
• Run the test with the provided command PYTORCH_TEST_WITH_DYNAMO=1 python test/test_sparse.py TestSparseCPU.test_sparse_addmm_cpu_float16 to reproduce the error and gather more information.
• Check the workflow logs and grep for test_sparse_addmm_cpu_float16 to find relevant log snippets and study the error messages.

Example

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

Notes

The error message AssertionError: compile_context is not set suggests that there is an issue with the PyTorch Dynamo compilation process, but the exact cause is not clear from the provided information. Further investigation and debugging are required to resolve the issue.

Recommendation

Apply workaround: Temporarily disable the PyTorch Dynamo compilation for the specific test or investigate alternative compilation methods until the root cause of the error is resolved.