Add complex to literal constant (#117819)

Summary: Fixes #ISSUE_NUMBER X-link: pytorch/pytorch#117819 Approved by: https://github.com/zou3519 Reviewed By: DanilBaibak Differential Revision: D53082842 Pulled By: yanboliang fbshipit-source-id: 22ed663ada1b8d95aadbc37bdeb0557a95daf702
pytorch · Jan 25, 2024 · 236d6bf · 236d6bf
1 parent 3f895ca
commit 236d6bf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
@@ -898,10 +898,12 @@ def rot_n_helper(n):
 common_constant_types = {
     int,
     float,
+    complex,
     bool,
     str,
     bytes,
     type(None),
+    Ellipsis.__class__,
     types.CodeType,
     torch.device,
     torch.dtype,