Enable torch.is_complex

fduwjj · fduwjj · commit 84a89882a03c · 2023-06-07T15:36:21.000Z
ghstack-source-id: 14a4125 Pull Request resolved: #103154
diff --git a/test/dynamo/test_functions.py b/test/dynamo/test_functions.py
@@ -399,6 +399,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
@@ -69,6 +69,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
`@@ -69,6 +69,7 @@`
`69`	`69`	`torch.is_autocast_cache_enabled,`
`70`	`70`	`torch.is_autocast_cpu_enabled,`
`71`	`71`	`torch.is_autocast_enabled,`
	`72`	`+ torch.is_complex,`
`72`	`73`	`torch.is_floating_point,`
`73`	`74`	`torch.nn.functional._Reduction.get_enum,`
`74`	`75`	`]`