Merge pull request #3982 from ReinUsesLisp/membar-cts

shader/other: Implement MEMBAR.CTS
This commit is contained in:
bunnei 2020-05-30 11:51:42 -04:00 committed by GitHub
commit 058ec22787
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 27 additions and 9 deletions

View File

@ -2344,7 +2344,12 @@ private:
return {}; return {};
} }
Expression MemoryBarrierGL(Operation) { Expression MemoryBarrierGroup(Operation) {
code.AddLine("groupMemoryBarrier();");
return {};
}
Expression MemoryBarrierGlobal(Operation) {
code.AddLine("memoryBarrier();"); code.AddLine("memoryBarrier();");
return {}; return {};
} }
@ -2591,7 +2596,8 @@ private:
&GLSLDecompiler::ShuffleIndexed, &GLSLDecompiler::ShuffleIndexed,
&GLSLDecompiler::Barrier, &GLSLDecompiler::Barrier,
&GLSLDecompiler::MemoryBarrierGL, &GLSLDecompiler::MemoryBarrierGroup,
&GLSLDecompiler::MemoryBarrierGlobal,
}; };
static_assert(operation_decompilers.size() == static_cast<std::size_t>(OperationCode::Amount)); static_assert(operation_decompilers.size() == static_cast<std::size_t>(OperationCode::Amount));

View File

@ -2215,8 +2215,8 @@ private:
return {}; return {};
} }
Expression MemoryBarrierGL(Operation) { template <spv::Scope scope>
const auto scope = spv::Scope::Device; Expression MemoryBarrier(Operation) {
const auto semantics = const auto semantics =
spv::MemorySemanticsMask::AcquireRelease | spv::MemorySemanticsMask::UniformMemory | spv::MemorySemanticsMask::AcquireRelease | spv::MemorySemanticsMask::UniformMemory |
spv::MemorySemanticsMask::WorkgroupMemory | spv::MemorySemanticsMask::WorkgroupMemory |
@ -2681,7 +2681,8 @@ private:
&SPIRVDecompiler::ShuffleIndexed, &SPIRVDecompiler::ShuffleIndexed,
&SPIRVDecompiler::Barrier, &SPIRVDecompiler::Barrier,
&SPIRVDecompiler::MemoryBarrierGL, &SPIRVDecompiler::MemoryBarrier<spv::Scope::Workgroup>,
&SPIRVDecompiler::MemoryBarrier<spv::Scope::Device>,
}; };
static_assert(operation_decompilers.size() == static_cast<std::size_t>(OperationCode::Amount)); static_assert(operation_decompilers.size() == static_cast<std::size_t>(OperationCode::Amount));

View File

@ -299,9 +299,19 @@ u32 ShaderIR::DecodeOther(NodeBlock& bb, u32 pc) {
break; break;
} }
case OpCode::Id::MEMBAR: { case OpCode::Id::MEMBAR: {
UNIMPLEMENTED_IF(instr.membar.type != Tegra::Shader::MembarType::GL);
UNIMPLEMENTED_IF(instr.membar.unknown != Tegra::Shader::MembarUnknown::Default); UNIMPLEMENTED_IF(instr.membar.unknown != Tegra::Shader::MembarUnknown::Default);
bb.push_back(Operation(OperationCode::MemoryBarrierGL)); const OperationCode type = [instr] {
switch (instr.membar.type) {
case Tegra::Shader::MembarType::CTA:
return OperationCode::MemoryBarrierGroup;
case Tegra::Shader::MembarType::GL:
return OperationCode::MemoryBarrierGlobal;
default:
UNIMPLEMENTED_MSG("MEMBAR type={}", static_cast<int>(instr.membar.type.Value()));
return OperationCode::MemoryBarrierGlobal;
}
}();
bb.push_back(Operation(type));
break; break;
} }
case OpCode::Id::DEPBAR: { case OpCode::Id::DEPBAR: {

View File

@ -234,7 +234,8 @@ enum class OperationCode {
ShuffleIndexed, /// (uint value, uint index) -> uint ShuffleIndexed, /// (uint value, uint index) -> uint
Barrier, /// () -> void Barrier, /// () -> void
MemoryBarrierGL, /// () -> void MemoryBarrierGroup, /// () -> void
MemoryBarrierGlobal, /// () -> void
Amount, Amount,
}; };