From 22a21995ad4a6c6eb34ceca61d6c43f4be2afd61 Mon Sep 17 00:00:00 2001
From: alex-weaver <awsweaver@gmail.com>
Date: Tue, 10 Jul 2018 02:56:24 +0100
Subject: [PATCH] [TVM] Fixed SPIR-V codegen for OpControlBarrier (#1409)
---
src/codegen/spirv/codegen_spirv.cc | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/src/codegen/spirv/codegen_spirv.cc b/src/codegen/spirv/codegen_spirv.cc
index 2c71d94ea..395bdff14 100644
--- a/src/codegen/spirv/codegen_spirv.cc
+++ b/src/codegen/spirv/codegen_spirv.cc
@@ -93,12 +93,14 @@ spirv::Value CodeGenSPIRV::CreateStorageSync(const Call* op) {
if (sync == "warp") {
return value;
} else if (sync == "shared") {
+ auto type_int = builder_->GetSType(Int(32));
builder_->MakeInst(
- spv::OpControlBarrier,
- spv::ScopeWorkgroup,
- spv::ScopeWorkgroup,
+ spv::OpControlBarrier,
+ builder_->IntImm(type_int, static_cast<int64_t>(spv::ScopeWorkgroup)),
+ builder_->IntImm(type_int, static_cast<int64_t>(spv::ScopeWorkgroup)),
+ builder_->IntImm(type_int, static_cast<int64_t>(
spv::MemorySemanticsSequentiallyConsistentMask |
- spv::MemorySemanticsWorkgroupMemoryMask);
+ spv::MemorySemanticsWorkgroupMemoryMask)));
} else {
LOG(FATAL) << "Do not support sync " << sync;
}
--
GitLab