Function testWaveMinMax

Source/Tools/FalcorTest/Tests/Slang/WaveOps.cpp:113–155

}

void testWaveMinMax(GPUUnitTestContext& ctx, bool conditional)
{
    ref<Device> pDevice = ctx.getDevice();

    DefineList defines = {{"CONDITIONAL", conditional ? "1" : "0"}};
    ctx.createProgram(kShaderFilename, "testWaveMinMax", defines, SlangCompilerFlags::None, ShaderModel::SM6_0);
    ctx.allocateStructuredBuffer("result", kNumElems * 2);

    auto var = ctx.vars().getRootVar();
    uint32_t zero = 0;
    auto pLaneCount = pDevice->createTypedBuffer<uint32_t>(1, ResourceBindFlags::UnorderedAccess, MemoryType::DeviceLocal, &zero);
    var["laneCount"] = pLaneCount;

    std::uniform_real_distribution<float> u(0.f, 1.f);
    std::vector<float> testData(kNumElems);
    for (size_t i = 0; i < testData.size(); i += 32)
    {
        float offset = 10.f * u(rng) - 5.f;
        for (size_t j = 0; j < 32; j++)
            testData[i + j] = offset + 2.f * u(rng) - 1.f;
    }
    var["testData"] = pDevice->createTypedBuffer<uint32_t>(
        kNumElems, ResourceBindFlags::ShaderResource, MemoryType::DeviceLocal, (uint32_t*)testData.data()
    );

    ctx.runProgram(kNumElems, 1, 1);

    // Get the lane count. We abort the test if it is an unsupported count.
    uint32_t laneCount = pLaneCount->getElement<uint32_t>(0);
    if (laneCount < 4 || laneCount > 128)
        FALCOR_THROW("Unsupported wave lane count");

    // Verify results of wave min/max.
    std::vector<float> expectedResult = computeMinMaxResult(testData, laneCount, conditional);
    FALCOR_ASSERT(expectedResult.size() == testData.size() * 2);

    std::vector<float4> result = ctx.readBuffer<float4>("result");
    for (size_t i = 0; i < testData.size(); i++)
    {
        EXPECT_EQ(result[2 * i + 0].x, expectedResult[2 * i + 0]) << "WaveActiveMin (i = " << i << ")";
        EXPECT_EQ(result[2 * i + 1].x, expectedResult[2 * i + 1]) << "WaveActiveMax (i = " << i << ")";
    }
}

uint32_t queryLaneCount(GPUUnitTestContext& ctx)
{