From e48925027a79fd6aaa02d1a00338df41dd505b1b Mon Sep 17 00:00:00 2001
From: Scott Wolchok <swolchok@meta.com>
Date: Mon, 27 Jan 2025 09:24:34 -0800
Subject: [PATCH] Add missing dtypes to gen_selected_op_variants.py

Summary: These were in scalar_type.h but not here. We should be able to understand all dtypes in this script even if they're not all implemented, because a custom op might support them.

Differential Revision: D68714101
---
 codegen/tools/gen_selected_op_variants.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/codegen/tools/gen_selected_op_variants.py b/codegen/tools/gen_selected_op_variants.py
index 661f1e0728b..da1c1215e2e 100644
--- a/codegen/tools/gen_selected_op_variants.py
+++ b/codegen/tools/gen_selected_op_variants.py
@@ -59,6 +59,13 @@
     "20": "Bits4x2",
     "21": "Bits8",
     "22": "Bits16",
+    "23": "Float8_e5m2",
+    "24": "Float8_e4m3fn",
+    "25": "Float8_e5m2fnuz",
+    "26": "Float8_e4m3fnuz",
+    "27": "UInt16",
+    "28": "UInt32",
+    "29": "Uint64",
 }