Get enum's underlying value type

vaithak · Aug 7, 2024 · 97acf5f · 97acf5f
1 parent 1f4e2a3
commit 97acf5f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/Differentiator/CladUtils.cpp b/lib/Differentiator/CladUtils.cpp
@@ -372,6 +372,10 @@ namespace clad {
       else if (T->isArrayType())
         valueType =
             T->getPointeeOrArrayElementType()->getCanonicalTypeInternal();
+      else if (T->isEnumeralType()){
+        if (const auto* ET = dyn_cast<EnumType>(T))
+          valueType = ET->getDecl()->getIntegerType();
+      }
       valueType.removeLocalConst();
       return valueType;
     }