Fix enum type literal assignment

vgvassilev · Oct 7, 2024 · a4f2ccb · a4f2ccb
1 parent 2878005
commit a4f2ccb
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lib/Differentiator/ConstantFolder.cpp b/lib/Differentiator/ConstantFolder.cpp
@@ -141,7 +141,16 @@ namespace clad {
     // SourceLocation noLoc;
     Expr* Result = 0;
     QT = QT.getCanonicalType();
-    if (QT->isPointerType()) {
+    if (QT->isEnumeralType()) {
+      llvm::APInt APVal(C.getIntWidth(QT), val,
+                        QT->isSignedIntegerOrEnumerationType());
+      Result = clad::synthesizeLiteral(
+          dyn_cast<EnumType>(QT)->getDecl()->getIntegerType(), C, APVal);
+      Expr* cast = ImplicitCastExpr::Create(
+          C, QT, clang::CastKind::CK_IntegralCast, Result, nullptr, VK_PRValue,
+          FPOptionsOverride());
+      Result = cast;
+    } else if (QT->isPointerType()) {
       Result = clad::synthesizeLiteral(QT, C);
     } else if (QT->isBooleanType()) {
       Result = clad::synthesizeLiteral(QT, C, (bool)val);