AstPrinter::emitStmt: reuse generic statementContext for regular statement

Sven Verdoolaege · Sven Verdoolaege · commit 96ebfa317869 · 2018-04-10T10:10:18.000+02:00
This removes some minor code duplication.
diff --git a/src/core/polyhedral/cuda/codegen.cc b/src/core/polyhedral/cuda/codegen.cc
@@ -424,12 +424,11 @@ void AstPrinter::emitStmt(isl::ast_node_user node) {
       stmtId.get_name() == kReadIdName || stmtId.get_name() == kWriteIdName) {
     emitCopyStmt(statementContext);
   } else { // regular statement
-    auto mappedStmtId =
-        context_.iteratorMaps.at(nodeId).get_tuple_id(isl::dim_type::out);
+    auto mappedStmtId = statementContext.statementId();
     CHECK_EQ(stmtId, mappedStmtId)
         << "statement ids in expr (" << stmtId << ") and in iteratorMaps ("
         << mappedStmtId << ") do not match";
-    emitUserStmt(stmtId, CodegenStatementContext(context_, nodeId));
+    emitUserStmt(stmtId, statementContext);
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -424,12 +424,11 @@ void AstPrinter::emitStmt(isl::ast_node_user node) {`
`424`	`424`	`stmtId.get_name() == kReadIdName \|\| stmtId.get_name() == kWriteIdName) {`
`425`	`425`	`emitCopyStmt(statementContext);`
`426`	`426`	`} else { // regular statement`
`427`		`- auto mappedStmtId =`
`428`		`- context_.iteratorMaps.at(nodeId).get_tuple_id(isl::dim_type::out);`
	`427`	`+ auto mappedStmtId = statementContext.statementId();`
`429`	`428`	`CHECK_EQ(stmtId, mappedStmtId)`
`430`	`429`	`<< "statement ids in expr (" << stmtId << ") and in iteratorMaps ("`
`431`	`430`	`<< mappedStmtId << ") do not match";`
`432`		`- emitUserStmt(stmtId, CodegenStatementContext(context_, nodeId));`
	`431`	`+ emitUserStmt(stmtId, statementContext);`
`433`	`432`	`}`
`434`	`433`	`}`
`435`	`434`