Don't pretend that we will IR print Stmts.

Author: mcourteaux

src/IRGraphCXXPrinter.cpp

@@ -40,6 +40,7 @@ std::string IRGraphCXXPrinter::to_cpp_arg<Expr>(const Expr &e) {
     return node_names.at(e.get());
 }
 
+// Not used, but leaving in place in case we ever want to expand this to Stmts.
 template<>
 std::string IRGraphCXXPrinter::to_cpp_arg<Stmt>(const Stmt &s) {
     if (!s.defined()) {
@@ -60,6 +61,8 @@ template<>
 std::string IRGraphCXXPrinter::to_cpp_arg<std::string>(const std::string &s) {
     return "\"" + s + "\"";
 }
+
+// Not used, but leaving in place in case we ever want to expand this to Stmts.
 template<>
 std::string IRGraphCXXPrinter::to_cpp_arg<ForType>(const ForType &f) {
     switch (f) {
@@ -210,6 +213,7 @@ VISIT_NODE(Call, op->type, op->name, op->args, op->call_type, op->func, op->valu
 VISIT_NODE(Let, op->name, op->value, op->body)
 VISIT_NODE(VectorReduce, op->op, op->value, op->type.lanes())
 
+#if 0  // Currently no support yet for Stmts, however, the macros below are already correct. We just can't print everything yet.
 // --- 6. Core Statements ---
 VISIT_NODE(LetStmt, op->name, op->value, op->body)
 VISIT_NODE(AssertStmt, op->condition, op->message)
@@ -232,6 +236,7 @@ VISIT_NODE(ProducerConsumer, op->name, op->is_producer, op->body)
 VISIT_NODE(Acquire, op->semaphore, op->count, op->body)
 VISIT_NODE(Fork, op->first, op->rest)
 VISIT_NODE(Atomic, op->producer_name, op->mutex_name, op->body)
+#endif
 
 void IRGraphCXXPrinter::test() {
     // This:

src/IRGraphCXXPrinter.h

@@ -30,12 +30,6 @@ class IRGraphCXXPrinter : public IRGraphVisitor {
         }
     }
 
-    void print(const Stmt &s) {
-        if (s.defined()) {
-            s.accept(this);
-        }
-    }
-
 private:
     template<typename T, typename... Args>
     void emit_node(const char *node_type_str, const T *op, Args &&...args);
@@ -47,6 +41,8 @@ class IRGraphCXXPrinter : public IRGraphVisitor {
     std::string to_cpp_arg(const std::vector<T> &vec);
 
 protected:
+    using IRGraphVisitor::visit;
+
     void visit(const IntImm *) override;
     void visit(const UIntImm *) override;
     void visit(const FloatImm *) override;
@@ -74,26 +70,9 @@ class IRGraphCXXPrinter : public IRGraphVisitor {
     void visit(const Ramp *) override;
     void visit(const Broadcast *) override;
     void visit(const Let *) override;
-    void visit(const LetStmt *) override;
-    void visit(const AssertStmt *) override;
-    void visit(const ProducerConsumer *) override;
-    void visit(const Store *) override;
-    void visit(const Provide *) override;
-    void visit(const Allocate *) override;
-    void visit(const Free *) override;
-    void visit(const Realize *) override;
-    void visit(const Block *) override;
-    void visit(const Fork *) override;
-    void visit(const IfThenElse *) override;
-    void visit(const Evaluate *) override;
     void visit(const Call *) override;
     void visit(const Variable *) override;
-    void visit(const For *) override;
-    void visit(const Acquire *) override;
     void visit(const Shuffle *) override;
-    void visit(const Prefetch *) override;
-    void visit(const HoistedStorage *) override;
-    void visit(const Atomic *) override;
     void visit(const VectorReduce *) override;
 
 public: