shader_decode: ATOM/ATOMS: add function to avoid code repetition
This commit is contained in:
parent
c8f6d9effd
commit
4f7bea403a
|
@ -19,7 +19,6 @@ namespace VideoCommon::Shader {
|
||||||
using Tegra::Shader::AtomicOp;
|
using Tegra::Shader::AtomicOp;
|
||||||
using Tegra::Shader::AtomicType;
|
using Tegra::Shader::AtomicType;
|
||||||
using Tegra::Shader::Attribute;
|
using Tegra::Shader::Attribute;
|
||||||
using Tegra::Shader::GlobalAtomicOp;
|
|
||||||
using Tegra::Shader::GlobalAtomicType;
|
using Tegra::Shader::GlobalAtomicType;
|
||||||
using Tegra::Shader::Instruction;
|
using Tegra::Shader::Instruction;
|
||||||
using Tegra::Shader::OpCode;
|
using Tegra::Shader::OpCode;
|
||||||
|
@ -28,6 +27,34 @@ using Tegra::Shader::StoreType;
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
|
Node getAtomOperation(AtomicOp op, bool is_signed, Node memory, Node data) {
|
||||||
|
switch (op) {
|
||||||
|
case AtomicOp::Add:
|
||||||
|
return SignedOperation(OperationCode::AtomicIAdd, is_signed, std::move(memory),
|
||||||
|
std::move(data));
|
||||||
|
case AtomicOp::Min:
|
||||||
|
return SignedOperation(OperationCode::AtomicIMin, is_signed, std::move(memory),
|
||||||
|
std::move(data));
|
||||||
|
case AtomicOp::Max:
|
||||||
|
return SignedOperation(OperationCode::AtomicIMax, is_signed, std::move(memory),
|
||||||
|
std::move(data));
|
||||||
|
case AtomicOp::And:
|
||||||
|
return SignedOperation(OperationCode::AtomicIAnd, is_signed, std::move(memory),
|
||||||
|
std::move(data));
|
||||||
|
case AtomicOp::Or:
|
||||||
|
return SignedOperation(OperationCode::AtomicIOr, is_signed, std::move(memory),
|
||||||
|
std::move(data));
|
||||||
|
case AtomicOp::Xor:
|
||||||
|
return SignedOperation(OperationCode::AtomicIXor, is_signed, std::move(memory),
|
||||||
|
std::move(data));
|
||||||
|
case AtomicOp::Exch:
|
||||||
|
return SignedOperation(OperationCode::AtomicIExchange, is_signed, std::move(memory),
|
||||||
|
std::move(data));
|
||||||
|
default:
|
||||||
|
return Immediate(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
bool IsUnaligned(Tegra::Shader::UniformType uniform_type) {
|
bool IsUnaligned(Tegra::Shader::UniformType uniform_type) {
|
||||||
return uniform_type == Tegra::Shader::UniformType::UnsignedByte ||
|
return uniform_type == Tegra::Shader::UniformType::UnsignedByte ||
|
||||||
uniform_type == Tegra::Shader::UniformType::UnsignedShort;
|
uniform_type == Tegra::Shader::UniformType::UnsignedShort;
|
||||||
|
@ -363,9 +390,9 @@ u32 ShaderIR::DecodeMemory(NodeBlock& bb, u32 pc) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
case OpCode::Id::ATOM: {
|
case OpCode::Id::ATOM: {
|
||||||
UNIMPLEMENTED_IF_MSG(instr.atom.operation == GlobalAtomicOp::Inc ||
|
UNIMPLEMENTED_IF_MSG(instr.atom.operation == AtomicOp::Inc ||
|
||||||
instr.atom.operation == GlobalAtomicOp::Dec ||
|
instr.atom.operation == AtomicOp::Dec ||
|
||||||
instr.atom.operation == GlobalAtomicOp::SafeAdd,
|
instr.atom.operation == AtomicOp::SafeAdd,
|
||||||
"operation={}", static_cast<int>(instr.atom.operation.Value()));
|
"operation={}", static_cast<int>(instr.atom.operation.Value()));
|
||||||
UNIMPLEMENTED_IF_MSG(instr.atom.type == GlobalAtomicType::S64 ||
|
UNIMPLEMENTED_IF_MSG(instr.atom.type == GlobalAtomicType::S64 ||
|
||||||
instr.atom.type == GlobalAtomicType::U64,
|
instr.atom.type == GlobalAtomicType::U64,
|
||||||
|
@ -381,36 +408,8 @@ u32 ShaderIR::DecodeMemory(NodeBlock& bb, u32 pc) {
|
||||||
const bool is_signed =
|
const bool is_signed =
|
||||||
instr.atoms.type == AtomicType::S32 || instr.atoms.type == AtomicType::S64;
|
instr.atoms.type == AtomicType::S32 || instr.atoms.type == AtomicType::S64;
|
||||||
Node gmem = MakeNode<GmemNode>(real_address, base_address, descriptor);
|
Node gmem = MakeNode<GmemNode>(real_address, base_address, descriptor);
|
||||||
Node data = GetRegister(instr.gpr20);
|
Node value = getAtomOperation(static_cast<AtomicOp>(instr.atom.operation), is_signed, gmem,
|
||||||
|
GetRegister(instr.gpr20));
|
||||||
Node value = [&]() {
|
|
||||||
switch (instr.atoms.operation) {
|
|
||||||
case AtomicOp::Add:
|
|
||||||
return SignedOperation(OperationCode::AtomicIAdd, is_signed, std::move(gmem),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Min:
|
|
||||||
return SignedOperation(OperationCode::AtomicIMin, is_signed, std::move(gmem),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Max:
|
|
||||||
return SignedOperation(OperationCode::AtomicIMax, is_signed, std::move(gmem),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::And:
|
|
||||||
return SignedOperation(OperationCode::AtomicIAnd, is_signed, std::move(gmem),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Or:
|
|
||||||
return SignedOperation(OperationCode::AtomicIOr, is_signed, std::move(gmem),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Xor:
|
|
||||||
return SignedOperation(OperationCode::AtomicIXor, is_signed, std::move(gmem),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Exch:
|
|
||||||
return SignedOperation(OperationCode::AtomicIExchange, is_signed, std::move(gmem),
|
|
||||||
std::move(data));
|
|
||||||
default:
|
|
||||||
UNREACHABLE();
|
|
||||||
return Immediate(0);
|
|
||||||
}
|
|
||||||
}();
|
|
||||||
SetRegister(bb, instr.gpr0, std::move(value));
|
SetRegister(bb, instr.gpr0, std::move(value));
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
@ -426,39 +425,9 @@ u32 ShaderIR::DecodeMemory(NodeBlock& bb, u32 pc) {
|
||||||
const s32 offset = instr.atoms.GetImmediateOffset();
|
const s32 offset = instr.atoms.GetImmediateOffset();
|
||||||
Node address = GetRegister(instr.gpr8);
|
Node address = GetRegister(instr.gpr8);
|
||||||
address = Operation(OperationCode::IAdd, std::move(address), Immediate(offset));
|
address = Operation(OperationCode::IAdd, std::move(address), Immediate(offset));
|
||||||
|
Node value =
|
||||||
Node memory = GetSharedMemory(std::move(address));
|
getAtomOperation(static_cast<AtomicOp>(instr.atoms.operation), is_signed,
|
||||||
Node data = GetRegister(instr.gpr20);
|
GetSharedMemory(std::move(address)), GetRegister(instr.gpr20));
|
||||||
|
|
||||||
Node value = [&]() {
|
|
||||||
switch (instr.atoms.operation) {
|
|
||||||
case AtomicOp::Add:
|
|
||||||
return SignedOperation(OperationCode::AtomicIAdd, is_signed, std::move(memory),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Min:
|
|
||||||
return SignedOperation(OperationCode::AtomicIMin, is_signed, std::move(memory),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Max:
|
|
||||||
return SignedOperation(OperationCode::AtomicIMax, is_signed, std::move(memory),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::And:
|
|
||||||
return SignedOperation(OperationCode::AtomicIAnd, is_signed, std::move(memory),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Or:
|
|
||||||
return SignedOperation(OperationCode::AtomicIOr, is_signed, std::move(memory),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Xor:
|
|
||||||
return SignedOperation(OperationCode::AtomicIXor, is_signed, std::move(memory),
|
|
||||||
std::move(data));
|
|
||||||
case AtomicOp::Exch:
|
|
||||||
return SignedOperation(OperationCode::AtomicIExchange, is_signed, std::move(memory),
|
|
||||||
std::move(data));
|
|
||||||
default:
|
|
||||||
UNREACHABLE();
|
|
||||||
return Immediate(0);
|
|
||||||
}
|
|
||||||
}();
|
|
||||||
|
|
||||||
SetRegister(bb, instr.gpr0, std::move(value));
|
SetRegister(bb, instr.gpr0, std::move(value));
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
@ -492,8 +461,8 @@ std::tuple<Node, Node, GlobalMemoryBase> ShaderIR::TrackGlobalMemory(NodeBlock&
|
||||||
|
|
||||||
const auto [base_address, index, offset] =
|
const auto [base_address, index, offset] =
|
||||||
TrackCbuf(addr_register, global_code, static_cast<s64>(global_code.size()));
|
TrackCbuf(addr_register, global_code, static_cast<s64>(global_code.size()));
|
||||||
ASSERT_OR_EXECUTE_MSG(base_address != nullptr,
|
ASSERT_OR_EXECUTE_MSG(
|
||||||
{ return std::make_tuple(nullptr, nullptr, GlobalMemoryBase{}); },
|
base_address != nullptr, { return std::make_tuple(nullptr, nullptr, GlobalMemoryBase{}); },
|
||||||
"Global memory tracking failed");
|
"Global memory tracking failed");
|
||||||
|
|
||||||
bb.push_back(Comment(fmt::format("Base address is c[0x{:x}][0x{:x}]", index, offset)));
|
bb.push_back(Comment(fmt::format("Base address is c[0x{:x}][0x{:x}]", index, offset)));
|
||||||
|
|
|
@ -86,6 +86,20 @@ OperationCode SignedToUnsignedCode(OperationCode operation_code, bool is_signed)
|
||||||
return OperationCode::LogicalUNotEqual;
|
return OperationCode::LogicalUNotEqual;
|
||||||
case OperationCode::LogicalIGreaterEqual:
|
case OperationCode::LogicalIGreaterEqual:
|
||||||
return OperationCode::LogicalUGreaterEqual;
|
return OperationCode::LogicalUGreaterEqual;
|
||||||
|
case OperationCode::AtomicIExchange:
|
||||||
|
return OperationCode::AtomicUExchange;
|
||||||
|
case OperationCode::AtomicIAdd:
|
||||||
|
return OperationCode::AtomicUAdd;
|
||||||
|
case OperationCode::AtomicIMin:
|
||||||
|
return OperationCode::AtomicUMin;
|
||||||
|
case OperationCode::AtomicIMax:
|
||||||
|
return OperationCode::AtomicUMax;
|
||||||
|
case OperationCode::AtomicIAnd:
|
||||||
|
return OperationCode::AtomicUAnd;
|
||||||
|
case OperationCode::AtomicIOr:
|
||||||
|
return OperationCode::AtomicUOr;
|
||||||
|
case OperationCode::AtomicIXor:
|
||||||
|
return OperationCode::AtomicUXor;
|
||||||
case OperationCode::INegate:
|
case OperationCode::INegate:
|
||||||
UNREACHABLE_MSG("Can't negate an unsigned integer");
|
UNREACHABLE_MSG("Can't negate an unsigned integer");
|
||||||
return {};
|
return {};
|
||||||
|
|
Reference in New Issue