[BugFix] shared.dyn support for codegen_nvptx

Signed-off-by: ZQPei <[email protected]>
apache · Nov 25, 2021 · a8d14b1 · a8d14b1
1 parent 4d6d4c7
commit a8d14b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/target/llvm/codegen_nvptx.cc b/src/target/llvm/codegen_nvptx.cc
@@ -138,7 +138,7 @@ class CodeGenNVPTX : public CodeGenLLVM {
     if (sync == "warp") {
       // TODO(tqchen) warp sync in CUDA9
       return nullptr;
-    } else if (sync == "shared") {
+    } else if (sync == "shared" || sync == "shared.dyn") {
       llvm::Function* f =
           llvm::Intrinsic::getDeclaration(module_.get(), ::llvm::Intrinsic::nvvm_barrier0);
       return builder_->CreateCall(f, {});