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,
 ]

Original file line number	Diff line number	Diff line change
`@@ -73,6 +73,7 @@`
`73`	`73`	`torch.is_autocast_cache_enabled,`
`74`	`74`	`torch.is_autocast_cpu_enabled,`
`75`	`75`	`torch.is_autocast_enabled,`
	`76`	`+ torch.is_complex,`
`76`	`77`	`torch.is_floating_point,`
`77`	`78`	`torch.nn.functional._Reduction.get_enum,`
`78`	`79`	`]`