Implement CallOpInterface for LLZK symbol lookups and fix up analyses accordingly

changelogs/unreleased/iangneal__analysis-tests.yaml

@@ -0,0 +1,6 @@
+added:
+  - PredecessorAnalysis (-llzk-predecessor-analysis) for testing
+changed:
+  - Implement `CallOpInterface` methods for `CallOp`
+removed:
+  - LLZK port of MLIR DenseAnalysis

include/llzk/Analysis/AnalysisPasses.h

@@ -26,6 +26,8 @@ std::unique_ptr<mlir::Pass> createSymbolDefTreePrinterPass();
 
 std::unique_ptr<mlir::Pass> createSymbolUseGraphPrinterPass();
 
+std::unique_ptr<mlir::Pass> createPredecessorPrinterPass();
+
 #define GEN_PASS_REGISTRATION
 #include "llzk/Analysis/AnalysisPasses.h.inc"
 

include/llzk/Analysis/AnalysisPasses.td

@@ -108,4 +108,10 @@ def SymbolUseGraphPrinterPass : LLZKPass<"llzk-print-symbol-use-graph"> {
   let options = [OutputStreamOption, SaveDotGraphOption];
 }
 
+def PredecessorPrinterPass : LLZKPass<"llzk-print-predecessors"> {
+  let summary = "Print the predecessors of all operations.";
+  let constructor = "llzk::createPredecessorPrinterPass()";
+  let options = [OutputStreamOption];
+}
+
 #endif // LLZK_ANALYSIS_TD

include/llzk/Analysis/AnalysisUtil.h

@@ -13,13 +13,10 @@
 
 namespace llzk::dataflow {
 
-/// LLZK: Added this utility to ensure analysis is performed for all structs
-/// in a given module.
-///
-/// @brief Mark all operations from the top and included in the top operation
-/// as live so the solver will perform dataflow analyses.
+/// @brief Loads analyses required to initialize the Executable and PredecessorState
+/// analysis states, which are required for the MLIR Dataflow analyses to properly
+/// traverse the LLZK IR.
 /// @param solver The solver.
-/// @param top The top-level operation.
-void markAllOpsAsLive(mlir::DataFlowSolver &solver, mlir::Operation *top);
+void loadRequiredAnalyses(mlir::DataFlowSolver &solver);
 
 } // namespace llzk::dataflow

include/llzk/Analysis/AnalysisWrappers.h

@@ -212,7 +212,7 @@ class ModuleAnalysis {
   /// in the `ModuleOp` that is being subjected to this analysis.
   /// @param am The module's analysis manager.
   void constructChildAnalyses(mlir::AnalysisManager &am) {
-    dataflow::markAllOpsAsLive(solver, modOp);
+    dataflow::loadRequiredAnalyses(solver);
 
     // The analysis is run at the module level so that lattices are computed
     // for global functions as well.

include/llzk/Analysis/ConstraintDependencyGraph.h

@@ -32,12 +32,13 @@ using SourceRefRemappings = std::vector<std::pair<SourceRef, SourceRefLatticeVal
 /// LLZK operations use and produce. The analysis is simple: any operation will
 /// simply output a union of its input references, regardless of what type of
 /// operation it performs, as the analysis is operator-insensitive.
-class SourceRefAnalysis : public dataflow::DenseForwardDataFlowAnalysis<SourceRefLattice> {
+class SourceRefAnalysis : public mlir::dataflow::DenseForwardDataFlowAnalysis<SourceRefLattice> {
 public:
-  using dataflow::DenseForwardDataFlowAnalysis<SourceRefLattice>::DenseForwardDataFlowAnalysis;
+  using mlir::dataflow::DenseForwardDataFlowAnalysis<
+      SourceRefLattice>::DenseForwardDataFlowAnalysis;
 
   void visitCallControlFlowTransfer(
-      mlir::CallOpInterface call, dataflow::CallControlFlowAction action,
+      mlir::CallOpInterface call, mlir::dataflow::CallControlFlowAction action,
       const SourceRefLattice &before, SourceRefLattice *after
   ) override;
 

include/llzk/Analysis/IntervalAnalysis.h

@@ -12,7 +12,6 @@
 #include "llzk/Analysis/AbstractLatticeValue.h"
 #include "llzk/Analysis/AnalysisWrappers.h"
 #include "llzk/Analysis/ConstraintDependencyGraph.h"
-#include "llzk/Analysis/DenseAnalysis.h"
 #include "llzk/Analysis/Intervals.h"
 #include "llzk/Analysis/SparseAnalysis.h"
 #include "llzk/Dialect/Array/IR/Ops.h"
@@ -26,6 +25,7 @@
 #include "llzk/Util/Compare.h"
 #include "llzk/Util/Field.h"
 
+#include <mlir/Analysis/DataFlow/DenseAnalysis.h>
 #include <mlir/IR/BuiltinOps.h>
 #include <mlir/Pass/AnalysisManager.h>
 #include <mlir/Support/LLVM.h>

include/llzk/Analysis/SourceRefLattice.h

@@ -10,10 +10,11 @@
 #pragma once
 
 #include "llzk/Analysis/AbstractLatticeValue.h"
-#include "llzk/Analysis/DenseAnalysis.h"
 #include "llzk/Analysis/SourceRef.h"
 #include "llzk/Util/ErrorHelper.h"
 
+#include <mlir/Analysis/DataFlow/DenseAnalysis.h>
+
 #include <llvm/ADT/PointerUnion.h>
 
 namespace llzk {
@@ -86,7 +87,7 @@ class SourceRefLatticeValue
 };
 
 /// A lattice for use in dense analysis.
-class SourceRefLattice : public dataflow::AbstractDenseLattice {
+class SourceRefLattice : public mlir::dataflow::AbstractDenseLattice {
 public:
   // mlir::Value is used for read-like operations that create references in their results,
   // mlir::Operation* is used for write-like operations that reference values as their destinations