[FIRRTL] Add BigIntConstantOp (llvm#5397)

adkian-sifive · Jun 13, 2023 · 87897b7 · 87897b7
1 parent 5271822
commit 87897b7
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 0 deletions.
diff --git a/include/circt/Dialect/FIRRTL/FIRRTLExpressions.td b/include/circt/Dialect/FIRRTL/FIRRTLExpressions.td
@@ -1009,6 +1009,21 @@ def StringConstantOp : FIRRTLOp<"string", [Pure, ConstantLike]> {
   let assemblyFormat = "$value attr-dict";
 }
 
+def BigIntConstantOp : FIRRTLOp<"bigint", [Pure, ConstantLike]> {
+  let summary = "Produce a constant bigint value";
+  let description = [{
+    Produces a constant value of bigint type.
+
+    Example:
+    ```mlir
+    %0 = firrtl.bigint 42
+    ```
+  }];
+  let arguments = (ins APSIntAttr:$value);
+  let results = (outs BigIntType:$result);
+  let hasCustomAssemblyFormat = true;
+}
+
 //===----------------------------------------------------------------------===//
 // RefOperations: Operations on the RefType.
 //

diff --git a/lib/Dialect/FIRRTL/FIRRTLOps.cpp b/lib/Dialect/FIRRTL/FIRRTLOps.cpp
@@ -3024,6 +3024,27 @@ Attribute AggregateConstantOp::getAttributeFromFieldID(uint64_t fieldID) {
   return value;
 }
 
+void BigIntConstantOp::print(OpAsmPrinter &p) {
+  p << " ";
+  p.printAttributeWithoutType(getValueAttr());
+  p.printOptionalAttrDict((*this)->getAttrs(), /*elidedAttrs=*/{"value"});
+}
+
+ParseResult BigIntConstantOp::parse(OpAsmParser &parser,
+                                    OperationState &result) {
+  auto *context = parser.getContext();
+  APInt value;
+  if (parser.parseInteger(value) ||
+      parser.parseOptionalAttrDict(result.attributes))
+    return failure();
+  result.addTypes(BigIntType::get(context));
+  auto intType =
+      IntegerType::get(context, value.getBitWidth(), IntegerType::Signed);
+  auto valueAttr = parser.getBuilder().getIntegerAttr(intType, value);
+  result.addAttribute("value", valueAttr);
+  return success();
+}
+
 LogicalResult BundleCreateOp::verify() {
   if (getType().getNumElements() != getFields().size())
     return emitOpError("number of fields doesn't match type");

diff --git a/test/Dialect/FIRRTL/test.mlir b/test/Dialect/FIRRTL/test.mlir
@@ -255,5 +255,10 @@ firrtl.module @StringTest(in %in: !firrtl.string, out %out: !firrtl.string) {
 // CHECK-SAME:  (in %in: !firrtl.bigint, out %out: !firrtl.bigint)
 firrtl.module @BigIntTest(in %in: !firrtl.bigint, out %out: !firrtl.bigint) {
   firrtl.propassign %out, %in : !firrtl.bigint
+
+  // CHECK: %0 = firrtl.bigint 4
+  %0 = firrtl.bigint 4
+  // CHECK: %1 = firrtl.bigint -4
+  %1 = firrtl.bigint -4
 }
 }