From 236d6bf8a117fcdd3a43fe467044d857ab59b703 Mon Sep 17 00:00:00 2001
From: "Yanbo Liang (Meta Employee)" <ybliang8@gmail.com>
Date: Thu, 25 Jan 2024 12:56:04 -0800
Subject: [PATCH] Add complex to literal constant (#117819)

Summary:
Fixes #ISSUE_NUMBER

X-link: https://github.com/pytorch/pytorch/pull/117819
Approved by: https://github.com/zou3519

Reviewed By: DanilBaibak

Differential Revision: D53082842

Pulled By: yanboliang

fbshipit-source-id: 22ed663ada1b8d95aadbc37bdeb0557a95daf702
---
 userbenchmark/dynamo/dynamobench/_dynamo/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
index 43dedb2b39..59e3c90503 100644
--- 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,