[Moore] Fix the stacking fault caused by cast and remove unused headers

include/circt/Dialect/Moore/MoorePasses.td

@@ -15,10 +15,10 @@
 
 include "mlir/Pass/PassBase.td"
 
-def SimplifyProcedures : Pass<"simplify-procedures", "moore::SVModuleOp"> {
+def SimplifyProcedures : Pass<"moore-simplify-procedures", "moore::SVModuleOp"> {
     let summary = "Simplify procedures";
     let description = [{
-      Cause we want to introduce mem2reg in the moore dialect to eliminate the
+      Because we want to introduce mem2reg in the moore dialect to eliminate the
       local temporary variables, if the local variabels exist in the procedure
       body, it can be promoted by mem2reg. But global/module-level variables
       don't be promoted. So this pass is aimed at inserting a local "shadow"

lib/Dialect/Moore/Transforms/SimplifyProcedures.cpp

@@ -14,11 +14,6 @@
 
 #include "circt/Dialect/Moore/MooreOps.h"
 #include "circt/Dialect/Moore/MoorePasses.h"
-#include "circt/Dialect/Moore/MooreTypes.h"
-#include "mlir/IR/Builders.h"
-#include "mlir/Pass/Pass.h"
-#include "llvm/Support/Casting.h"
-#include "llvm/Support/raw_ostream.h"
 
 namespace circt {
 namespace moore {
@@ -61,23 +56,27 @@ void SimplifyProceduresPass::runOnOperation() {
           if (!users.contains(user))
             users.insert(user);
 
-        auto varOp = cast<VariableOp>(nestedOp.getOperand(0).getDefiningOp());
-        auto varName = builder.getStringAttr("local_" + varOp.getName());
-        auto resultType = varOp.getResult().getType();
-        builder.setInsertionPointToStart(procedureOp.getBody());
-        auto readOp = builder.create<ReadOp>(
-            nestedOp.getLoc(), cast<RefType>(resultType).getNestedType(),
-            varOp.getResult());
-        auto newVarOp = builder.create<VariableOp>(nestedOp.getLoc(),
-                                                   resultType, varName, readOp);
-        builder.clearInsertionPoint();
+        // Because the operand of moore.event_wait is net.
+        if (auto varOp = llvm::dyn_cast_or_null<VariableOp>(
+                nestedOp.getOperand(0).getDefiningOp())) {
+          auto varName =
+              builder.getStringAttr(Twine("local_") + varOp.getName());
+          auto resultType = varOp.getResult().getType();
+          builder.setInsertionPointToStart(procedureOp.getBody());
+          auto readOp = builder.create<ReadOp>(
+              nestedOp.getLoc(), cast<RefType>(resultType).getNestedType(),
+              varOp.getResult());
+          auto newVarOp = builder.create<VariableOp>(
+              nestedOp.getLoc(), resultType, varName, readOp);
+          builder.clearInsertionPoint();
 
-        // Replace the users of the global variable with a corresponding
-        // "shadow" variable.
-        for (auto *user : users) {
-          user->replaceUsesOfWith(user->getOperand(0), newVarOp);
-          if (isa<BlockingAssignOp>(user))
-            assignOps.insert(user);
+          // Replace the users of the global variable with a corresponding
+          // "shadow" variable.
+          for (auto *user : users) {
+            user->replaceUsesOfWith(user->getOperand(0), newVarOp);
+            if (isa<BlockingAssignOp>(user))
+              assignOps.insert(user);
+          }
         }
       }
 

test/Dialect/Moore/simplify-procedures.mlir

@@ -1,4 +1,4 @@
-// RUN: circt-opt --simplify-procedures %s | FileCheck %s
+// RUN: circt-opt --moore-simplify-procedures %s | FileCheck %s
 
 // CHECK-LABEL: moore.module @Foo()
 moore.module @Foo() {