diff --git a/src/target/codegen_cuda.cc b/src/target/codegen_cuda.cc index 6b5f5063c..dda969253 100644 --- a/src/target/codegen_cuda.cc +++ b/src/target/codegen_cuda.cc @@ -1645,10 +1645,7 @@ void CodeGenTileLangCUDA::VisitExpr_(const CallNode *op, std::ostream &os) { } else if (op->op.same_as(tl::sync_grid())) { this->need_cooperative_groups_ = true; this->PrintIndent(); - this->stream << "cooperative_groups::grid_group grid = " - "cooperative_groups::this_grid();\n"; - this->PrintIndent(); - this->stream << "grid.sync();\n"; + this->stream << "cooperative_groups::this_grid().sync();\n"; } else if (op->op.same_as(tl::loop_break())) { this->PrintIndent(); this->stream << "break;\n";