Enable torch.is_complex in Dynamo tracing (#103154)

fduwjj · pytorchmergebot · commit b5021ba9817c · 2023-06-07T20:56:46.000Z
Pull Request resolved: #103154 Approved by: https://github.com/yanboliang
diff --git a/test/dynamo/test_functions.py b/test/dynamo/test_functions.py
@@ -436,6 +436,13 @@ def test_get_default_dtype(x):
         else:
             return x - 1
 
+    @make_test
+    def test_is_complex(x):
+        if torch.is_complex(x):
+            return x + 1
+        else:
+            return x - 1
+
     @make_test
     def test_device(x):
         if not x.is_cuda:
diff --git a/torch/_dynamo/variables/torch.py b/torch/_dynamo/variables/torch.py
@@ -73,6 +73,7 @@
     torch.is_autocast_cache_enabled,
     torch.is_autocast_cpu_enabled,
     torch.is_autocast_enabled,
+    torch.is_complex,
     torch.is_floating_point,
     torch.nn.functional._Reduction.get_enum,
 ]
