[RTG] Add ElaborationPass (#7876)

docs/Dialects/RTG.md

@@ -273,3 +273,7 @@ companion dialect to define any backends.
 ## Types
 
 [include "Dialects/RTGTypes.md"]
+
+## Passes
+
+[include "RTGPasses.md"]

include/circt/Dialect/RTG/CMakeLists.txt

@@ -1 +1,2 @@
 add_subdirectory(IR)
+add_subdirectory(Transforms)

include/circt/Dialect/RTG/Transforms/CMakeLists.txt

@@ -0,0 +1,6 @@
+set(LLVM_TARGET_DEFINITIONS RTGPasses.td)
+mlir_tablegen(RTGPasses.h.inc -gen-pass-decls)
+add_public_tablegen_target(CIRCTRTGTransformsIncGen)
+
+# Generate Pass documentation.
+add_circt_doc(RTGPasses RTGPasses -gen-pass-doc)

include/circt/Dialect/RTG/Transforms/RTGPasses.h

@@ -0,0 +1,31 @@
+//===- RTGPasses.h - RTG pass entry points ----------------------*- C++ -*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+// This header file defines prototypes that expose pass constructors.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef CIRCT_DIALECT_RTG_TRANSFORMS_RTGPASSES_H
+#define CIRCT_DIALECT_RTG_TRANSFORMS_RTGPASSES_H
+
+#include "mlir/Pass/Pass.h"
+#include <memory>
+
+namespace circt {
+namespace rtg {
+
+/// Generate the code for registering passes.
+#define GEN_PASS_REGISTRATION
+#define GEN_PASS_DECL
+#include "circt/Dialect/RTG/Transforms/RTGPasses.h.inc"
+#undef GEN_PASS_REGISTRATION
+
+} // namespace rtg
+} // namespace circt
+
+#endif // CIRCT_DIALECT_RTG_TRANSFORMS_RTGPASSES_H

include/circt/Dialect/RTG/Transforms/RTGPasses.td

@@ -0,0 +1,35 @@
+//===-- RTGPasses.td - RTG pass definition file ------------*- tablegen -*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+// This file defines the passes that operate on the RTG dialect.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef CIRCT_DIALECT_RTG_TRANSFORMS_RTGPASSES_TD
+#define CIRCT_DIALECT_RTG_TRANSFORMS_RTGPASSES_TD
+
+include "mlir/Pass/PassBase.td"
+
+def ElaborationPass : Pass<"rtg-elaborate", "mlir::ModuleOp"> {
+  let summary = "elaborate the randomization parts";
+  let description = [{
+    This pass interprets most RTG operations to perform the represented
+    randomization and in the process get rid of those operations. This means,
+    after this pass the IR does not contain any random constructs within tests
+    anymore.
+  }];
+
+  let options = [
+    Option<"seed", "seed", "unsigned", /*default=*/"",
+      "The seed for any RNG constructs used in the pass.">,
+  ];
+
+  let dependentDialects = ["mlir::arith::ArithDialect"];
+}
+
+#endif // CIRCT_DIALECT_RTG_TRANSFORMS_RTGPASSES_TD

include/circt/Dialect/RTGTest/IR/RTGTestOps.td

@@ -33,3 +33,13 @@ def CPUDeclOp : RTGTestOp<"cpu_decl", [
 
   let assemblyFormat = "$id attr-dict";
 }
+
+def ConstantTestOp : RTGTestOp<"constant_test", [
+  Pure, ConstantLike,
+]> {
+  let arguments = (ins AnyAttr:$value);
+  let results = (outs AnyType:$result);
+
+  let assemblyFormat = "type($result) attr-dict";
+  let hasFolder = 1;
+}

include/circt/InitAllPasses.h

@@ -32,6 +32,7 @@
 #include "circt/Dialect/Moore/MoorePasses.h"
 #include "circt/Dialect/OM/OMPasses.h"
 #include "circt/Dialect/Pipeline/PipelinePasses.h"
+#include "circt/Dialect/RTG/Transforms/RTGPasses.h"
 #include "circt/Dialect/SSP/SSPPasses.h"
 #include "circt/Dialect/SV/SVPasses.h"
 #include "circt/Dialect/Seq/SeqPasses.h"
@@ -73,6 +74,7 @@ inline void registerAllPasses() {
   sv::registerPasses();
   handshake::registerPasses();
   kanagawa::registerPasses();
+  rtg::registerPasses();
   hw::registerPasses();
   pipeline::registerPasses();
   sim::registerPasses();

lib/Dialect/RTG/CMakeLists.txt

@@ -1 +1,2 @@
 add_subdirectory(IR)
+add_subdirectory(Transforms)

lib/Dialect/RTG/Transforms/CMakeLists.txt

@@ -0,0 +1,16 @@
+add_circt_dialect_library(CIRCTRTGTransforms
+  ElaborationPass.cpp
+
+  DEPENDS
+  CIRCTRTGTransformsIncGen
+
+  LINK_COMPONENTS
+  Support
+
+  LINK_LIBS PRIVATE
+  CIRCTRTGDialect
+  MLIRArithDialect
+  MLIRIR
+  MLIRPass
+)
+