Skip to content

Commit a6191b9

Browse files
committed
[Codegen, CUDA] Enable emitting SyncWarp
1 parent 5fb4910 commit a6191b9

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/target/source/codegen_cuda.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -618,7 +618,8 @@ void CodeGenCUDA::PrintVecElemStore(const std::string& vec, DataType t, int i,
618618
void CodeGenCUDA::PrintStorageSync(const CallNode* op) {
619619
const std::string& sync = op->args[0].as<StringImmNode>()->value;
620620
if (sync == "warp") {
621-
// DO nothing.
621+
this->PrintIndent();
622+
this->stream << "__syncwarp();\n";
622623
} else if (sync == "shared" || sync == "shared.dyn") {
623624
this->PrintIndent();
624625
this->stream << "__syncthreads();\n";

0 commit comments

Comments
 (0)