File tree Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Original file line number Diff line number Diff line change @@ -2284,20 +2284,18 @@ void CodeGenTileLangCUDA::VisitStmt_(const AllocateNode *op) {
22842284 this ->PrintStmt (op->body );
22852285}
22862286
2287- void CodeGenTileLangCUDA::PrintDeviceAssert (const CallNode* call) {
2287+ void CodeGenTileLangCUDA::PrintDeviceAssert (const CallNode * call) {
22882288 std::string cond = PrintExpr (call->args [0 ]);
22892289 if (call->args .size () == 2 ) {
22902290 if (auto str = call->args [1 ].as <StringImmNode>()) {
22912291 this ->PrintIndent ();
2292- stream << " if (!(" << cond << " )) printf(\" "
2293- << str->value << " \\ n\" );\n " ;
2292+ stream << " if (!(" << cond << " )) printf(\" " << str->value << " \\ n\" );\n " ;
22942293 }
22952294 }
22962295 this ->PrintIndent ();
22972296 stream << " assert(" << cond << " );\n " ;
22982297}
22992298
2300-
23012299void CodeGenTileLangCUDA::VisitStmt_ (const EvaluateNode *op) {
23022300 if (is_const_int (op->value ))
23032301 return ;
@@ -2311,7 +2309,7 @@ void CodeGenTileLangCUDA::VisitStmt_(const EvaluateNode *op) {
23112309 stream << " " << vid_global_barrier_expect_ << " = 0;\n " ;
23122310 PrintIndent ();
23132311 stream << " }\n " ;
2314- }
2312+ }
23152313 if (call && call->op .as <OpNode>()) {
23162314 auto name = call->op .as <OpNode>()->name ;
23172315 if (name == " tl.device_assert" || name == " tl.device_assert_with_msg" ) {
Original file line number Diff line number Diff line change @@ -132,7 +132,7 @@ class CodeGenTileLangCUDA final : public CodeGenC {
132132 CodeGenTileLangCUDA *p);
133133 void PrintWmmaScope (const std::string &scope, DataType t,
134134 const VarNode *variable, std::ostream &os);
135- void PrintDeviceAssert (const CallNode* call);
135+ void PrintDeviceAssert (const CallNode * call);
136136 int32_t GetWmmaFragmentSize (const std::string &scope, const VarNode *variable,
137137 int32_t size);
138138
You can’t perform that action at this time.
0 commit comments