[export] Graph break on nn.Parameter construction

[ghstack-poisoned]
pytorch · Jun 18, 2024 · c223eac · c223eac
1 parent f7eae27
commit c223eac
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch/_dynamo/variables/torch.py b/torch/_dynamo/variables/torch.py
@@ -877,6 +877,9 @@ def handle_ntuple(value):
     @classmethod
     def call_nn_parameter(cls, tx, data=None, requires_grad=True):
         """A call to torch.nn.Parameter() gets lifted to before the graph"""
+        if tx.export:
+            unimplemented("nn parameter construction not supported with export")
+
         if isinstance(requires_grad, variables.VariableTracker):
             try:
                 requires_grad = requires_grad.as_python_constant()