Support wrapping Real types

Author: Ansley Adelaide Ussery

tests/test_modules.py

@@ -196,6 +196,7 @@ class NNModuleTests(torchdynamo.testing.TestCase):
     test_iseval2 = make_test(IsEvalCheck())
     test_viamodulecall = make_test(ViaModuleCall())
     test_isnonelayer = make_test(IsNoneLayer())
+    test_intarg = make_test(IntArg())
 
     # not yet implemented
     # test_layerlist = make_test(LayerList())

torchdynamo/symbolic_convert.py

@@ -4,6 +4,7 @@
 import functools
 import inspect
 import itertools
+from numbers import Real
 import operator
 import types
 import typing
@@ -178,7 +179,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:
+        elif value is True or value is False or value is None or isinstance(value, Real):
             # For these, just specialize on exact value
             return ConstantVariable(
                 value=value,