111 lines
5.3 KiB
C#
111 lines
5.3 KiB
C#
// Copyright (C) 2026 SharpEmu Emulator Project
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
using System.Buffers.Binary;
|
|
using SharpEmu.HLE;
|
|
using SharpEmu.Libs.Gpu.Buffers;
|
|
using SharpEmu.Libs.Tests.Gpu.Images;
|
|
using SharpEmu.Libs.Tests.Gpu.Scheduling;
|
|
using SharpEmu.ShaderCompiler;
|
|
using SharpEmu.ShaderCompiler.Resources;
|
|
using SharpEmu.ShaderCompiler.Tests.Resources;
|
|
using SharpEmu.ShaderCompiler.Vulkan;
|
|
using Xunit;
|
|
using Xunit.Abstractions;
|
|
|
|
namespace SharpEmu.Libs.Tests.Gpu.Vulkan;
|
|
|
|
[Collection(SchedulingStateCollection.Name)]
|
|
public sealed class UnsignedDifferenceShaderTests(HeadlessVulkanFixture fixture, ITestOutputHelper output)
|
|
: IClassFixture<HeadlessVulkanFixture>
|
|
{
|
|
private const ulong ShaderAddress = 0x1_0000_0000;
|
|
private const ulong BufferAddress = 0x2_0000_0000;
|
|
private const uint Sentinel = 0xCAFE_BABE;
|
|
private const int RecordBytes = 16;
|
|
|
|
[Fact]
|
|
public void SumOfAbsoluteDifferences_MatchesUnsignedBoundaryResultsOnTheDevice()
|
|
{
|
|
var vulkan = fixture.Vulkan;
|
|
if (!GatePrerequisites.Ready(vulkan, shaderInt64: true)) return;
|
|
|
|
uint[] operands = [0, 1, 0x7FFF_FFFF, 0x8000_0000, 0xFFFF_FFFE, uint.MaxValue];
|
|
uint[] accumulators = [0, 0x8000_0000, uint.MaxValue];
|
|
var recordCount = operands.Length * operands.Length * accumulators.Length;
|
|
var inputBytes = new byte[recordCount * RecordBytes];
|
|
var expectedBytes = new byte[inputBytes.Length];
|
|
var recordIndex = 0;
|
|
foreach (var left in operands)
|
|
foreach (var right in operands)
|
|
foreach (var accumulator in accumulators)
|
|
{
|
|
var record = inputBytes.AsSpan(recordIndex * RecordBytes, RecordBytes);
|
|
BinaryPrimitives.WriteUInt32LittleEndian(record, left);
|
|
BinaryPrimitives.WriteUInt32LittleEndian(record[4..], right);
|
|
BinaryPrimitives.WriteUInt32LittleEndian(record[8..], accumulator);
|
|
BinaryPrimitives.WriteUInt32LittleEndian(record[12..], Sentinel);
|
|
record.CopyTo(expectedBytes.AsSpan(recordIndex * RecordBytes));
|
|
var expected = unchecked((uint)(Math.Abs((long)left - right) + accumulator));
|
|
BinaryPrimitives.WriteUInt32LittleEndian(expectedBytes.AsSpan(recordIndex * RecordBytes + 12), expected);
|
|
recordIndex++;
|
|
}
|
|
|
|
var request = CreateBoundaryRequest((uint)recordCount);
|
|
Assert.True(Gen5SpirvTranslator.TryCompileProgram(request, out var shader, out var error), error);
|
|
using (var harness = new ImageTestHarness(vulkan))
|
|
using (var runner = new LayoutComputeRunner(harness, request, shader.Spirv))
|
|
{
|
|
var records = runner.CreateBuffer(inputBytes);
|
|
uint[] userData = [(uint)(BufferAddress & uint.MaxValue), (uint)(BufferAddress >> 32), (uint)inputBytes.Length, 0];
|
|
var bindings = new Dictionary<DescriptorBindingKind, GpuBuffer[]> { [DescriptorBindingKind.Buffers] = [records] };
|
|
harness.Run(() => runner.Dispatch(userData, bindings, 1));
|
|
var actualBytes = harness.ReadBack(records.Handle, 0, (ulong)inputBytes.Length);
|
|
Assert.Equal(expectedBytes, actualBytes);
|
|
harness.AssertNoValidationMessages();
|
|
}
|
|
|
|
vulkan.AssertNoValidationMessages();
|
|
output.WriteLine($"Verified {recordCount} unsigned boundary cases on {vulkan.DeviceName}; validation={vulkan.ValidationEnabled}.");
|
|
}
|
|
|
|
private static ShaderCompileRequest CreateBoundaryRequest(uint threadCount)
|
|
{
|
|
var memory = new FakeCpuMemory(ShaderAddress, 0x1000);
|
|
uint[] words = [0xD15D_000F, 0x10Cu | (0x10Du << 9) | (0x10Eu << 18), 0xBF81_0000];
|
|
var code = new byte[words.Length * sizeof(uint)];
|
|
for (var wordIndex = 0; wordIndex < words.Length; wordIndex++)
|
|
{
|
|
BinaryPrimitives.WriteUInt32LittleEndian(code.AsSpan(wordIndex * sizeof(uint)), words[wordIndex]);
|
|
}
|
|
|
|
Assert.True(memory.TryWrite(ShaderAddress, code));
|
|
Assert.True(Gen5ShaderTranslator.TryDecodeProgram(new CpuContext(memory, Generation.Gen5), ShaderAddress,
|
|
out var decoded, out var decodeError), decodeError);
|
|
var addition = decoded.Instructions[0];
|
|
Assert.Equal("VSadU32", addition.Opcode);
|
|
Assert.Equal([Gen5Operand.Vector(12), Gen5Operand.Vector(13), Gen5Operand.Vector(14)], addition.Sources);
|
|
Assert.Equal([Gen5Operand.Vector(15)], addition.Destinations);
|
|
|
|
var program = new Gen5ShaderProgram(ShaderAddress,
|
|
[
|
|
new(0, Gen5ShaderEncoding.Vop2, "VLshlrevB32", [0],
|
|
[Gen5Operand.Source(132), Gen5Operand.Vector(0)], [Gen5Operand.Vector(10)], null),
|
|
new(4, Gen5ShaderEncoding.Mubuf, "BufferLoadDwordx3", [0, 0], [], [],
|
|
new Gen5BufferMemoryControl(3, 10, 12, 0, 0, false, true, false, false)),
|
|
addition with { Pc = 12 },
|
|
new(20, Gen5ShaderEncoding.Mubuf, "BufferStoreDword", [0, 0], [], [],
|
|
new Gen5BufferMemoryControl(1, 10, 15, 0, 12, false, true, false, false)),
|
|
decoded.Instructions[1] with { Pc = 28 },
|
|
]);
|
|
var (plan, resources, layout) = ResourceTestProgram.Prepare(program, userDataCount: 4);
|
|
return new ShaderCompileRequest(plan, resources, layout)
|
|
{
|
|
LocalSizeX = threadCount,
|
|
ThreadCountX = threadCount,
|
|
ThreadCountY = 1,
|
|
ThreadCountZ = 1,
|
|
};
|
|
}
|
|
}
|