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

Traceback (most recent call last): File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/testing/_internal/common_utils.py", line 3528, in wrapper method(*args, **kwargs) ~~~~~~^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/testing/_internal/common_device_type.py", line 515, in instantiated_test result = test(self, **param_kwargs) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/testing/_internal/common_utils.py", line 1735, in wrapper fn(*args, **kwargs) ~~^^^^^^^^^^^^^^^^^ File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 2506, 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 2217, 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 772, in call result = _compile( frame.f_code, ...<18 lines>... convert_frame_box=self._box, ) File "/opt/conda/envs/py_3.14/lib/python3.14t/site-packages/torch/_dynamo/convert_frame.py", line 2001, 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 1611, 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 1670, 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 1512, 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_mask_backward_cpu_float64

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_mask_backward_cpu_float64 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 test case TestSparseCPU.test_sparse_mask_backward_cpu_float64 to understand the specific conditions that are causing the failure.
• Run the test with the provided command PYTORCH_TEST_WITH_DYNAMO=1 python test/test_sparse.py TestSparseCPU.test_sparse_mask_backward_cpu_float64 to reproduce the error and gather more information.
• Check the workflow logs and grep for test_sparse_mask_backward_cpu_float64 to find relevant log snippets and study the error messages.

Example

No code snippet is provided as the issue is related to a specific test case and the error message is an AssertionError that requires further investigation.

Notes

The error message suggests that the compile_context is not being set, which is a critical component of the PyTorch Dynamo compilation process. Without more information, it is difficult to provide a specific fix. Further investigation and debugging are required to resolve this issue.

Recommendation

Apply workaround: Investigate and resolve the AssertionError: compile_context is not set issue in the PyTorch Dynamo compilation process. This may involve modifying the test case or the compilation process to ensure that the compile_context is properly set.