Check for none for NNModuleVariable.__module__ (#93326)

Summary: Pull Request resolved: #93326

Test Plan: CI

Differential Revision: D42869182

fbshipit-source-id: 153788f46f6b504ee9c5d0623510d017f0789ed1

torch/_dynamo/variables/functions.py

@@ -278,14 +278,12 @@ def python_type(self):
     def call_function(
         self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
     ) -> "VariableTracker":
-        if (
-            isinstance(self.obj, variables.NNModuleVariable)
-            and getattr(self.fn, "__module__", "").startswith("torch.nn.")
-            or self.is_constant
-        ):
-            return self.obj.call_method(
-                tx, self.fn.__name__, args, kwargs, constant=self.is_constant
-            ).add_options(self)
+        if isinstance(self.obj, variables.NNModuleVariable):
+            module_attr = getattr(self.fn, "__module__", "")
+            if module_attr is not None and module_attr.startswith("torch.nn.") or self.is_constant:
+                return self.obj.call_method(
+                    tx, self.fn.__name__, args, kwargs, constant=self.is_constant
+                ).add_options(self)
         return super().call_function(tx, args, kwargs)
 
     def num_parameters(self):