diff --git a/src/StmtToViz.cpp b/src/StmtToViz.cpp index ae498530d5f9..f4f06fda650f 100644 --- a/src/StmtToViz.cpp +++ b/src/StmtToViz.cpp @@ -2080,10 +2080,12 @@ class HTMLCodePrinter : public IRVisitor { } void visit(const VectorReduce *op) override { - print_opening_tag("div", "VectorReduce"); + print_opening_tag("span", "VectorReduce"); + print_text("("); print_type(op->type); + print_text(")"); print_function_call("vector_reduce", {op->op, op->value}); - print_closing_tag("div"); + print_closing_tag("span"); print_ln(); }