Support wrapping Real types

Author: Ansley Adelaide Ussery

torchdynamo/symbolic_convert.py

@@ -171,6 +171,14 @@ def wrap_local(self, name, value):
                 state=TracingSupported.YES,
                 guards={Guard(name, GuardSource.LOCAL, GuardBuilder.TYPE_MATCH)},
             )
+        elif isinstance(value, Real):
+            self.graphargs.append(LocalArg(name))
+            self.create_graph_input(name)
+            return ConstantVariable(
+                value=value,
+                state=TracingSupported.UNKNOWN,
+                guards={Guard(name, GuardSource.LOCAL, GuardBuilder.VALUE_MATCH)},
+            )
         elif isinstance(value, torch.nn.Module):
             key = f"{name}_{next(self.cnt)}"
             self.nn_modules[key] = value
@@ -179,7 +187,7 @@ def wrap_local(self, name, value):
                 state=TracingSupported.YES,
                 guards={Guard(name, GuardSource.LOCAL, GuardBuilder.VALUE_MATCH)},
             )
-        elif value is True or value is False or value is None or isinstance(value, Real):
+        elif value is True or value is False or value is None:
             # For these, just specialize on exact value
             return ConstantVariable(
                 value=value,