diff --git a/src/graphics/guest_gpu/command_processor/pm4Handlers.cpp b/src/graphics/guest_gpu/command_processor/pm4Handlers.cpp index 557302d2e..cd0247cd0 100644 --- a/src/graphics/guest_gpu/command_processor/pm4Handlers.cpp +++ b/src/graphics/guest_gpu/command_processor/pm4Handlers.cpp @@ -3644,6 +3644,12 @@ void GraphicsInitJmpTablesShIndirect() { g_hw_sh_indirect_func[Pm4::SPI_SHADER_USER_DATA_ADDR_HI_GS] = [](KYTY_HW_SH_INDIRECT_ARGS) { HwShIgnoreShaderRegister(cmd_offset, value); }; + g_hw_sh_indirect_func[Pm4::SPI_SHADER_PGM_CHKSUM_HS] = [](KYTY_HW_SH_INDIRECT_ARGS) { + HwShIgnoreShaderRegister(cmd_offset, value); + }; + g_hw_sh_indirect_func[Pm4::SPI_SHADER_PGM_RSRC4_HS] = [](KYTY_HW_SH_INDIRECT_ARGS) { + HwShIgnoreShaderRegister(cmd_offset, value); + }; g_hw_sh_indirect_func[Pm4::SPI_GRAPHICS_SHADER_CONTROL_HS] = [](KYTY_HW_SH_INDIRECT_ARGS) { HwShIgnoreShaderRegister(cmd_offset, value); }; diff --git a/src/graphics/guest_gpu/pm4.h b/src/graphics/guest_gpu/pm4.h index 7cce9ff63..0da1f65ad 100644 --- a/src/graphics/guest_gpu/pm4.h +++ b/src/graphics/guest_gpu/pm4.h @@ -834,8 +834,10 @@ constexpr uint32_t SPI_SHADER_USER_ACCUM_ESGS_0 = 0xB2; constexpr uint32_t SPI_SHADER_USER_ACCUM_ESGS_3 = 0xB5; constexpr uint32_t SPI_SHADER_PGM_LO_ES = 0xC8; constexpr uint32_t SPI_SHADER_PGM_HI_ES = 0xC9; -// Retained only as an opaque AGC compiler-record tag; 0x100 is not a PS5 SH register. +// Private AGC registers omitted from the public ShRegOffset enum. Shader binaries can still +// include them in the ShRegister arrays submitted through SET_SH_REG_INDIRECT. constexpr uint32_t SPI_SHADER_PGM_CHKSUM_HS = 0x100; +constexpr uint32_t SPI_SHADER_PGM_RSRC4_HS = 0x101; constexpr uint32_t SPI_SHADER_USER_DATA_ADDR_LO_HS = 0x102; constexpr uint32_t SPI_SHADER_USER_DATA_ADDR_HI_HS = 0x103; constexpr uint32_t SPI_GRAPHICS_SHADER_CONTROL_HS = 0x107; diff --git a/tests/ShaderRecompilerComputeTests.cpp b/tests/ShaderRecompilerComputeTests.cpp index 09b72c5d6..a93eaac6e 100644 --- a/tests/ShaderRecompilerComputeTests.cpp +++ b/tests/ShaderRecompilerComputeTests.cpp @@ -24556,10 +24556,10 @@ void CheckPm4NativeTargetGeometryRegisters(RenderContext &renderer) { // Removed GCN shader resource/checksum/queue registers. Numeric offsets keep // this check independent of the deleted legacy names. - constexpr std::array legacy_shader_slots{ + constexpr std::array legacy_shader_slots{ 0x000u, 0x001u, 0x002u, 0x003u, 0x030u, 0x081u, 0x0b0u, 0x0bcu, - 0x100u, 0x101u, 0x130u, 0x14au, 0x14bu, 0x20eu, 0x20fu, 0x210u, - 0x211u, 0x216u, 0x217u, 0x219u, 0x21au, 0x27du, + 0x130u, 0x14au, 0x14bu, 0x20eu, 0x20fu, 0x210u, 0x211u, 0x216u, + 0x217u, 0x219u, 0x21au, 0x27du, }; for (const auto offset : legacy_shader_slots) { legacy_slots_are_unhandled &= g_hw_sh_func[offset] == nullptr && @@ -24601,6 +24601,30 @@ void CheckPm4NativeTargetGeometryRegisters(RenderContext &renderer) { std::printf("[host] %-32s ok\n", "Pm4NativeTargetGeometry"); } +void CheckPm4PrivateAgcShaderRegisters(RenderContext &renderer) { + GraphicsInitJmpTables(); + CommandProcessor processor(renderer, 0); + std::array registers{ + Pm4::SPI_SHADER_PGM_CHKSUM_HS, 0x12345678u, + Pm4::SPI_SHADER_PGM_RSRC4_HS, 0x87654321u, + }; + const auto address = reinterpret_cast(registers.data()); + std::array command{ + KYTY_PM4(5, Pm4::IT_SET_SH_REG_INDIRECT, Pm4::R_ZERO), + static_cast(address), static_cast(address >> 32u), + 0x80000000u, 2u, + }; + Pm4Execution execution; + const bool handlers_present = + g_hw_sh_indirect_func[Pm4::SPI_SHADER_PGM_CHKSUM_HS] != nullptr && + g_hw_sh_indirect_func[Pm4::SPI_SHADER_PGM_RSRC4_HS] != nullptr; + Require("Pm4PrivateAgcShaderRegisters", "private HS register stream", + handlers_present && + processor.Process(execution, command) == Pm4ProcessResult::Complete, + "private AGC HS registers were rejected by SET_SH_REG_INDIRECT"); + std::printf("[host] %-32s ok\n", "Pm4PrivateAgcShaderRegisters"); +} + void CheckPm4DirectShaderRegisterFallback(RenderContext &renderer) { GraphicsInitJmpTables(); CommandProcessor processor(renderer, 0); @@ -25361,6 +25385,7 @@ int main(int argc, char **argv) { if (argc == 2 && std::strcmp(argv[1], "--context-state-only") == 0) { VulkanHarness vulkan; CheckPm4NativeTargetGeometryRegisters(vulkan.RuntimeRenderer()); + CheckPm4PrivateAgcShaderRegisters(vulkan.RuntimeRenderer()); CheckPm4BlendColorRegisterRanges(vulkan.RuntimeRenderer()); CheckPm4PolygonOffsetRegisters(vulkan.RuntimeRenderer()); CheckPm4ContextStateOperations(vulkan.RuntimeRenderer()); @@ -25541,6 +25566,7 @@ int main(int argc, char **argv) { CheckPm4SyntheticOcclusionCounterDump(vulkan.RuntimeRenderer()); CheckPm4StencilInfoValueLane(vulkan.RuntimeRenderer()); CheckPm4NativeTargetGeometryRegisters(vulkan.RuntimeRenderer()); + CheckPm4PrivateAgcShaderRegisters(vulkan.RuntimeRenderer()); CheckPm4DirectShaderRegisterFallback(vulkan.RuntimeRenderer()); CheckPm4GuardBandRegisterRanges(vulkan.RuntimeRenderer()); CheckPm4BlendColorRegisterRanges(vulkan.RuntimeRenderer());