Skip to content

Commit 307778e

Browse files
committed
format
1 parent c94fe1c commit 307778e

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

src/target/codegen_cuda.cc

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff 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-
23012299
void 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") {

src/target/codegen_cuda.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)