Allow constant-folding of arbitrary-precision integer casts

p4lang · Jan 10, 2024 · 76af275 · 76af275
1 parent d7dfee1
commit 76af275
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/frontends/common/constantFolding.cpp b/frontends/common/constantFolding.cpp
@@ -808,6 +808,16 @@ const IR::Node *DoConstantFolding::postorder(IR::Cast *e) {
         } else {
             return e;
         }
+    } else if (etype->is<IR::Type_InfInt>()) {
+        if (const auto *constant = expr->to<IR::Constant>()) {
+            const auto *ctype = constant->type;
+            if (!ctype->is<IR::Type_Bits>() && !ctype->is<IR::Type_InfInt>()) {
+                ::error(ErrorType::ERR_INVALID, "%1%: Cannot cast %1% to arbitrary presion integer",
+                        ctype);
+                return e;
+            }
+            return new IR::Constant(e->srcInfo, etype, constant->value, constant->base);
+        }
     } else if (etype->is<IR::Type_Boolean>()) {
         if (expr->is<IR::BoolLiteral>()) return expr;
         if (expr->is<IR::Constant>()) {