multiple-output spat computes

src/PIM/Conversion/SpatialToGraphviz/SpatialToGraphviz.cpp

@@ -42,15 +42,15 @@ private:
   raw_ostream& os;
 
   /**
-   * Draws the subgraph for a given spatial::SpatWeightedCompute, including:
+   * Draws the subgraph for a given spatial::SpatCompute, including:
    * 1. Input nodes (block arguments)
    * 2. Operations
    * 3. Edges between yield (output) and its users
    *
-   * @param op The spatial::SpatWeightedCompute to draw the subgraph for.
+   * @param op The spatial::SpatCompute to draw the subgraph for.
    * @param computeNum The number of the compute operation.
    */
-  void drawComputeOpSubgraph(spatial::SpatWeightedCompute op, size_t computeNum) {
+  void drawComputeOpSubgraph(spatial::SpatCompute op, size_t computeNum) {
     os << "\tsubgraph cluster" << computeNum << " {\n\t\tlabel=\"Compute" << computeNum << "\";\n"
        << "\t\tstyle=filled;\n"
        << "\t\tcolor=lightblue;\n";
@@ -217,7 +217,7 @@ void SpatialToGraphvizPass::runOnOperation() {
   // 1. Print their subgraph
   // 2. Print the edges from its inputs to its outputs
   for (Operation& op : func.getOps()) {
-    if (auto computeOp = dyn_cast<spatial::SpatWeightedCompute>(op)) {
+    if (auto computeOp = dyn_cast<spatial::SpatCompute>(op)) {
       drawComputeOpSubgraph(computeOp, computeNum++);
     }
     else if (auto concatOp = dyn_cast<tensor::ConcatOp>(op)) {