mirror of
https://github.com/KytyPS5/KytyPS5.git
synced 2026-10-21 19:36:55 +02:00
Nawid3333
e93b4b0378
* shader: support DS_CMPST_RTN_B32 Assassin's Creed Valhalla (PPSA01490, #281) stops on DS opcode 0x30 in a compute shader. DS compares DATA and stores DATA2, the opposite of BUFFER_ATOMIC_CMPSWAP, so the 32-bit atomic path swaps the two for this opcode, as the 64-bit form does. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> * shader: reject unsupported GDS compare-store variant --------- Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com> Co-authored-by: nmzik <Nmzik@mail.ru>
257 lines
9.1 KiB
C++
257 lines
9.1 KiB
C++
#include "graphics/shader/recompiler/ir/opcodes/ValueOpcodes.h"
|
|
|
|
#include <array>
|
|
|
|
namespace Libs::Graphics::ShaderRecompiler::IR {
|
|
namespace {
|
|
|
|
struct OpcodeMeta {
|
|
std::string_view name;
|
|
Type type = Type::Void;
|
|
std::array<Type, 16> args = {};
|
|
size_t num_args = 0;
|
|
};
|
|
|
|
template <typename... Args>
|
|
consteval OpcodeMeta MakeMeta(std::string_view name, Type type, Args... args) {
|
|
static_assert(sizeof...(Args) <= 16);
|
|
OpcodeMeta meta {.name = name, .type = type, .num_args = sizeof...(Args)};
|
|
meta.args.fill(Type::Void);
|
|
size_t index = 0;
|
|
((meta.args[index++] = args), ...);
|
|
return meta;
|
|
}
|
|
|
|
constexpr Type Void = Type::Void;
|
|
constexpr Type Opaque = Type::Opaque;
|
|
constexpr Type ScalarReg = Type::ScalarReg;
|
|
constexpr Type VectorReg = Type::VectorReg;
|
|
constexpr Type U1 = Type::U1;
|
|
constexpr Type U8 = Type::U8;
|
|
constexpr Type U16 = Type::U16;
|
|
constexpr Type U32 = Type::U32;
|
|
constexpr Type U64 = Type::U64;
|
|
constexpr Type F16 = Type::F16;
|
|
constexpr Type F32 = Type::F32;
|
|
constexpr Type F64 = Type::F64;
|
|
constexpr Type U32x2 = Type::U32x2;
|
|
constexpr Type U32x3 = Type::U32x3;
|
|
constexpr Type U32x4 = Type::U32x4;
|
|
constexpr Type F32x2 = Type::F32x2;
|
|
constexpr Type SrtResource = Type::SrtResource;
|
|
constexpr Type BufferResource = Type::BufferResource;
|
|
constexpr Type AddressResource = Type::AddressResource;
|
|
constexpr Type ImageResource = Type::ImageResource;
|
|
constexpr Type SamplerResource = Type::SamplerResource;
|
|
constexpr Type ImageAddress = Type::ImageAddress;
|
|
|
|
constexpr std::array<OpcodeMeta, static_cast<size_t>(ValueOpcode::Count)> MetaTable = {{
|
|
#define VALUE_OPCODE(name, ...) MakeMeta(#name __VA_OPT__(, ) __VA_ARGS__),
|
|
#include "graphics/shader/recompiler/ir/opcodes/ValueOpcodes.inc"
|
|
#undef VALUE_OPCODE
|
|
}};
|
|
|
|
} // namespace
|
|
|
|
Type TypeOf(ValueOpcode opcode) {
|
|
return MetaTable[static_cast<size_t>(opcode)].type;
|
|
}
|
|
|
|
size_t NumArgsOf(ValueOpcode opcode) {
|
|
return MetaTable[static_cast<size_t>(opcode)].num_args;
|
|
}
|
|
|
|
Type ArgTypeOf(ValueOpcode opcode, size_t index) {
|
|
const auto& meta = MetaTable[static_cast<size_t>(opcode)];
|
|
return meta.args.at(index);
|
|
}
|
|
|
|
std::string_view ValueOpcodeName(ValueOpcode opcode) {
|
|
return MetaTable[static_cast<size_t>(opcode)].name;
|
|
}
|
|
|
|
bool HasSideEffects(ValueOpcode opcode) {
|
|
const auto buffer_access = BufferAccessOf(opcode);
|
|
if (buffer_access == BufferAccess::Write || buffer_access == BufferAccess::Atomic) {
|
|
return true;
|
|
}
|
|
const auto shared_access = SharedAccessOf(opcode);
|
|
if (shared_access == SharedAccess::Write || shared_access == SharedAccess::Atomic ||
|
|
shared_access == SharedAccess::Append || shared_access == SharedAccess::Consume) {
|
|
return true;
|
|
}
|
|
if (AddressOpcodeInfoOf(opcode).access == AddressAccess::Write) {
|
|
return true;
|
|
}
|
|
const auto image_info = ImageOpcodeInfoOf(opcode);
|
|
if (image_info.access == ImageAccess::Write || image_info.access == ImageAccess::Atomic) {
|
|
return true;
|
|
}
|
|
switch (opcode) {
|
|
case ValueOpcode::Reference:
|
|
case ValueOpcode::ReferenceU32:
|
|
case ValueOpcode::SetAttribute:
|
|
case ValueOpcode::SetTessellationAttribute:
|
|
case ValueOpcode::MeshAllocate:
|
|
case ValueOpcode::StoreCompletion:
|
|
case ValueOpcode::Barrier: return true;
|
|
default: return false;
|
|
}
|
|
}
|
|
|
|
BufferAccess BufferAccessOf(ValueOpcode opcode) {
|
|
switch (opcode) {
|
|
case ValueOpcode::ReadConstBuffer:
|
|
case ValueOpcode::LoadBufferU8:
|
|
case ValueOpcode::LoadBufferU16:
|
|
case ValueOpcode::LoadBufferU32:
|
|
case ValueOpcode::LoadBufferU32x2:
|
|
case ValueOpcode::LoadBufferU32x3:
|
|
case ValueOpcode::LoadBufferU32x4: return BufferAccess::Read;
|
|
case ValueOpcode::StoreBufferU8:
|
|
case ValueOpcode::StoreBufferU16:
|
|
case ValueOpcode::StoreBufferU32:
|
|
case ValueOpcode::StoreBufferU32x2:
|
|
case ValueOpcode::StoreBufferU32x3:
|
|
case ValueOpcode::StoreBufferU32x4: return BufferAccess::Write;
|
|
case ValueOpcode::BufferAtomicSwap32:
|
|
case ValueOpcode::BufferAtomicCmpSwap32:
|
|
case ValueOpcode::BufferAtomicSwap64:
|
|
case ValueOpcode::BufferAtomicIAdd32:
|
|
case ValueOpcode::BufferAtomicIAdd64:
|
|
case ValueOpcode::BufferAtomicISub32:
|
|
case ValueOpcode::BufferAtomicSMin32:
|
|
case ValueOpcode::BufferAtomicUMin32:
|
|
case ValueOpcode::BufferAtomicSMax32:
|
|
case ValueOpcode::BufferAtomicUMax32:
|
|
case ValueOpcode::BufferAtomicAnd32:
|
|
case ValueOpcode::BufferAtomicAnd64:
|
|
case ValueOpcode::BufferAtomicOr32:
|
|
case ValueOpcode::BufferAtomicOr64:
|
|
case ValueOpcode::BufferAtomicXor32:
|
|
case ValueOpcode::BufferAtomicFMin32:
|
|
case ValueOpcode::BufferAtomicFMax32: return BufferAccess::Atomic;
|
|
default: return BufferAccess::None;
|
|
}
|
|
}
|
|
|
|
uint32_t BufferComponentCount(ValueOpcode opcode) {
|
|
switch (opcode) {
|
|
case ValueOpcode::BufferAtomicSwap64:
|
|
case ValueOpcode::BufferAtomicIAdd64:
|
|
case ValueOpcode::BufferAtomicAnd64:
|
|
case ValueOpcode::BufferAtomicOr64:
|
|
case ValueOpcode::LoadBufferU32x2:
|
|
case ValueOpcode::StoreBufferU32x2: return 2u;
|
|
case ValueOpcode::LoadBufferU32x3:
|
|
case ValueOpcode::StoreBufferU32x3: return 3u;
|
|
case ValueOpcode::LoadBufferU32x4:
|
|
case ValueOpcode::StoreBufferU32x4: return 4u;
|
|
default: return BufferAccessOf(opcode) == BufferAccess::None ? 0u : 1u;
|
|
}
|
|
}
|
|
|
|
SharedAccess SharedAccessOf(ValueOpcode opcode) {
|
|
switch (opcode) {
|
|
case ValueOpcode::LoadSharedU8:
|
|
case ValueOpcode::LoadSharedU16:
|
|
case ValueOpcode::LoadSharedU32:
|
|
case ValueOpcode::LoadSharedU32x2:
|
|
case ValueOpcode::LoadSharedU32x3:
|
|
case ValueOpcode::LoadSharedU32x4: return SharedAccess::Read;
|
|
case ValueOpcode::WriteSharedU8:
|
|
case ValueOpcode::WriteSharedU16:
|
|
case ValueOpcode::WriteSharedU32:
|
|
case ValueOpcode::WriteSharedU32x2:
|
|
case ValueOpcode::WriteSharedU32x3:
|
|
case ValueOpcode::WriteSharedU32x4: return SharedAccess::Write;
|
|
case ValueOpcode::SharedAtomicFMin32:
|
|
case ValueOpcode::SharedAtomicFMax32:
|
|
case ValueOpcode::SharedAtomicSwap32:
|
|
case ValueOpcode::SharedAtomicIAdd32:
|
|
case ValueOpcode::SharedAtomicIAdd64:
|
|
case ValueOpcode::SharedAtomicISub32:
|
|
case ValueOpcode::SharedAtomicInc32:
|
|
case ValueOpcode::SharedAtomicDec32:
|
|
case ValueOpcode::SharedAtomicSMin32:
|
|
case ValueOpcode::SharedAtomicUMin32:
|
|
case ValueOpcode::SharedAtomicSMax32:
|
|
case ValueOpcode::SharedAtomicUMax32:
|
|
case ValueOpcode::SharedAtomicUMax64:
|
|
case ValueOpcode::SharedAtomicCmpSwap32:
|
|
case ValueOpcode::SharedAtomicCmpSwap64:
|
|
case ValueOpcode::SharedAtomicAnd32:
|
|
case ValueOpcode::SharedAtomicOr32:
|
|
case ValueOpcode::SharedAtomicOr64:
|
|
case ValueOpcode::SharedAtomicXor32:
|
|
case ValueOpcode::SharedAtomicMaskedOr32: return SharedAccess::Atomic;
|
|
case ValueOpcode::DataAppend: return SharedAccess::Append;
|
|
case ValueOpcode::DataConsume: return SharedAccess::Consume;
|
|
default: return SharedAccess::None;
|
|
}
|
|
}
|
|
|
|
uint32_t SharedComponentCount(ValueOpcode opcode) {
|
|
switch (opcode) {
|
|
case ValueOpcode::SharedAtomicIAdd64:
|
|
case ValueOpcode::SharedAtomicOr64:
|
|
case ValueOpcode::SharedAtomicUMax64:
|
|
case ValueOpcode::SharedAtomicCmpSwap64:
|
|
case ValueOpcode::LoadSharedU32x2:
|
|
case ValueOpcode::WriteSharedU32x2: return 2u;
|
|
case ValueOpcode::LoadSharedU32x3:
|
|
case ValueOpcode::WriteSharedU32x3: return 3u;
|
|
case ValueOpcode::LoadSharedU32x4:
|
|
case ValueOpcode::WriteSharedU32x4: return 4u;
|
|
default: return SharedAccessOf(opcode) == SharedAccess::None ? 0u : 1u;
|
|
}
|
|
}
|
|
|
|
AddressOpcodeInfo AddressOpcodeInfoOf(ValueOpcode opcode) {
|
|
switch (opcode) {
|
|
case ValueOpcode::LoadAddressU8: return {AddressAccess::Read, 8u};
|
|
case ValueOpcode::LoadAddressU16: return {AddressAccess::Read, 16u};
|
|
case ValueOpcode::LoadAddressU32: return {AddressAccess::Read, 32u};
|
|
case ValueOpcode::StoreAddressU8: return {AddressAccess::Write, 8u};
|
|
case ValueOpcode::StoreAddressU16: return {AddressAccess::Write, 16u};
|
|
case ValueOpcode::StoreAddressU32: return {AddressAccess::Write, 32u};
|
|
default: return {};
|
|
}
|
|
}
|
|
|
|
ImageOpcodeInfo ImageOpcodeInfoOf(ValueOpcode opcode) {
|
|
switch (opcode) {
|
|
case ValueOpcode::ImageQueryDimensions:
|
|
case ValueOpcode::ImageRead: return {ImageAccess::Read, ImageResourceClass::Sampled, false};
|
|
case ValueOpcode::ImageQueryLod:
|
|
case ValueOpcode::ImageSampleRaw:
|
|
case ValueOpcode::ImageGatherRaw:
|
|
return {ImageAccess::Read, ImageResourceClass::Sampled, true};
|
|
case ValueOpcode::ImageWrite:
|
|
return {ImageAccess::Write, ImageResourceClass::Storage, false};
|
|
case ValueOpcode::ImageAtomicCompareSwap32:
|
|
case ValueOpcode::ImageAtomicSwap32:
|
|
case ValueOpcode::ImageAtomicSwap64:
|
|
case ValueOpcode::ImageAtomicIAdd32:
|
|
case ValueOpcode::ImageAtomicIAdd64:
|
|
case ValueOpcode::ImageAtomicSMin32:
|
|
case ValueOpcode::ImageAtomicUMin32:
|
|
case ValueOpcode::ImageAtomicUMin64:
|
|
case ValueOpcode::ImageAtomicSMax32:
|
|
case ValueOpcode::ImageAtomicUMax32:
|
|
case ValueOpcode::ImageAtomicUMax64:
|
|
case ValueOpcode::ImageAtomicAnd32:
|
|
case ValueOpcode::ImageAtomicAnd64:
|
|
case ValueOpcode::ImageAtomicOr32:
|
|
case ValueOpcode::ImageAtomicOr64:
|
|
case ValueOpcode::ImageAtomicXor32:
|
|
case ValueOpcode::ImageAtomicXor64:
|
|
case ValueOpcode::ImageAtomicFMin32:
|
|
case ValueOpcode::ImageAtomicFMax32:
|
|
return {ImageAccess::Atomic, ImageResourceClass::Storage, false};
|
|
default: return {};
|
|
}
|
|
}
|
|
|
|
} // namespace Libs::Graphics::ShaderRecompiler::IR
|