ZIR-280: Keccak2 InitCycle is deterministic 🎉

zirgen/Conversions/Typing/ZhlComponent.cpp

@@ -549,6 +549,16 @@ Zhlt::ComponentOp LoweringImpl::gen(ComponentOp component,
   auto ctor = builder.create<Zhlt::ComponentOp>(
       loc, mangledName, valueType, constructArgsTypes, layoutType);
   ctor.getBody().takeBody(body);
+
+  for (NamedAttribute attr : component->getDiscardableAttrs()) {
+    StringRef name = attr.getName();
+    if (name == "function" || name == "argument" || name == "generic" || name == "picus") {
+      ctor->setAttr(name, attr.getValue());
+    } else {
+      ctor->emitError() << "unknown attribute `" << name << "`";
+    }
+  }
+
   return ctor;
 }
 

zirgen/Dialect/ZHLT/IR/Dialect.cpp

@@ -37,10 +37,6 @@ class ZhltInlinerInterface : public DialectInlinerInterface {
   using DialectInlinerInterface::DialectInlinerInterface;
 
   bool isLegalToInline(Operation* call, Operation* callable, bool wouldBeCloned) const final {
-    if (isa<ComponentOp>(callable))
-      // Only inline aspects that have been generated for components, not for the original
-      // components.
-      return false;
     return isa<FunctionOpInterface>(callable);
   }
 

zirgen/Dialect/ZHLT/IR/TypeUtils.cpp

@@ -40,7 +40,7 @@ std::string mangledTypeName(StringRef componentName, llvm::ArrayRef<Attribute> t
       } else if (auto intAttr = dyn_cast<PolynomialAttr>(typeArg)) {
         stream << intAttr[0];
       } else if (auto intAttr = dyn_cast<IntegerAttr>(typeArg)) {
-        stream << intAttr;
+        stream << intAttr.getUInt();
       } else {
         llvm::errs() << "Mangling type " << typeArg << " not implemented\n";
         assert(false && "not implemented");

zirgen/compiler/picus/BUILD.bazel

@@ -0,0 +1,18 @@
+package(
+    default_visibility = ["//visibility:public"],
+)
+
+cc_library(
+    name = "picus",
+    srcs = [
+        "picus.cpp",
+    ],
+    hdrs = [
+        "picus.h",
+    ],
+    deps = [
+        "//zirgen/Dialect/ZHLT/IR",
+        "//zirgen/Dialect/ZStruct/IR",
+        "//zirgen/dsl/passes:passes",
+    ],
+)