Replace unimplemented with unimplemented_v2' in codegen.py` (#148069)

zeshengzong · pytorchmergebot · commit 1add61c24280 · 2025-03-06T20:42:37.000Z
Fixes #147913 - replace `unimplemented` in `codegen.py` - remove unused import `unimplemented` Pull Request resolved: #148069 Approved by: https://github.com/Skylion007, https://github.com/williamwen42
diff --git a/torch/_dynamo/codegen.py b/torch/_dynamo/codegen.py
@@ -36,7 +36,7 @@
     create_rot_n,
     Instruction,
 )
-from .exc import IncorrectUsage, unimplemented, unimplemented_v2
+from .exc import IncorrectUsage, unimplemented_v2
 from .source import AttrSource, ChainedSource, DictGetItemSource, Source
 from .utils import is_safe_constant, rot_n_helper
 from .variables.base import ValueMutationExisting, VariableTracker
@@ -197,7 +197,12 @@ def __call__(self, value, allow_cache=True):
             try:
                 self.call_reconstruct(source)
             except NotImplementedError:
-                unimplemented(f"reconstruct: {source}")
+                unimplemented_v2(
+                    gb_type="Reconstruction failure: source.reconstruct not implemented",
+                    context=str(source),
+                    explanation=f"Dynamo has no bytecode reconstruction implemented for {type(source)} variable {source}.",
+                    hints=[*graph_break_hints.DYNAMO_BUG],
+                )
 
             self._output.append(create_dup_top())
             self.add_cache(source)
