From 2c4f9810327d58023e9ec44f579cd78adf56317b Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Mon, 1 Apr 2024 15:56:02 -0700
Subject: Allow bit operators on enum types. (#3862)

* Allow bit operators on enum types.

* Fix.
---
 source/slang/slang-check-conversion.cpp | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'source/slang/slang-check-conversion.cpp')

diff --git a/source/slang/slang-check-conversion.cpp b/source/slang/slang-check-conversion.cpp
index e55a88077..7dcc8975c 100644
--- a/source/slang/slang-check-conversion.cpp
+++ b/source/slang/slang-check-conversion.cpp
@@ -894,6 +894,25 @@ namespace Slang
             }
             return true;
         }
+
+        // A enum type can be converted into its underlying tag type.
+        if (auto enumDecl = isEnumType(fromType))
+        {
+            Type* tagType = enumDecl->tagType;
+            if (tagType == toType)
+            {
+                if (outCost)
+                {
+                    *outCost = kConversionCost_RankPromotion;
+                }
+                if (outToExpr)
+                {
+                    *outToExpr = fromExpr;
+                }
+                return true;
+            }
+        }
+
         // matrix type with different layouts are convertible
         if (auto fromMatrixType = as<MatrixExpressionType>(fromType))
         {
-- 
cgit v1.2.3