chromium/third_party/dawn/src/dawn/tests/perf_tests/ShaderRobustnessPerf.cpp

// Copyright 2021 The Dawn & Tint Authors
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// 1. Redistributions of source code must retain the above copyright notice, this
//    list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice,
//    this list of conditions and the following disclaimer in the documentation
//    and/or other materials provided with the distribution.
//
// 3. Neither the name of the copyright holder nor the names of its
//    contributors may be used to endorse or promote products derived from
//    this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
// OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#include <sstream>
#include <string>
#include <vector>

#include "dawn/tests/perf_tests/DawnPerfTest.h"
#include "dawn/utils/WGPUHelpers.h"

namespace dawn {
namespace {

constexpr uint32_t kTileSize = …;
constexpr uint32_t kWorkgroupSizeX = …;
constexpr uint32_t kWorkgroupSizeY = …;
static_assert …;
static_assert …;

std::string GenMatMulFloatHeader() { … }

const std::string& kMatMulFloatSharedArray1D = …;
const std::string& kMatMulFloatSharedArray2D = …;
const std::string& kMatMulFloatBodyPart1 = …;
const std::string& kMatMulFloatBodyPart2Array1D = …;
const std::string& kMatMulFloatBodyPart2Array2D = …;
const std::string& kMatMulFloatBodyPart3 = …;
std::string GenMatMulFloatOneDimensionalSharedArray() { … }

std::string GenMatMulFloatTwoDimensionalSharedArray() { … }

// The vec4 version requires that dimInner and dimBOuter are divisible by 4.
std::string GenMatMulVec4Header() { … }
const std::string& kMatMulVec4SharedArray1D = …;
const std::string& kMatMulVec4SharedArray2D = …;
const std::string& kMatMulVec4BodyPart1 = …;
const std::string& kMatMulVec4BodyPart2Array1D = …;
const std::string& kMatMulVec4BodyPart2Array2D = …;
const std::string& kMatMulVec4BodyPart3 = …;

std::string GenMatMulVec4OneDimensionalSharedArray() { … }

std::string GenMatMulVec4TwoDimensionalSharedArray() { … }

constexpr unsigned int kNumIterations = …;

enum class MatMulMethod { … };

std::ostream& operator<<(std::ostream& ostream, const MatMulMethod& matMulMethod) { … }

enum class ElemType { … };

std::ostream& operator<<(std::ostream& ostream, const ElemType& elemType) { … }

DimAOuter;
DimInner;
DimBOuter;
DAWN_TEST_PARAM_STRUCT(ShaderRobustnessParams,
                       MatMulMethod,
                       ElemType,
                       DimAOuter,
                       DimInner,
                       DimBOuter);

// Test the execution time of matrix multiplication (A [dimAOuter, dimInner] * B [dimInner,
// dimBOuter]) on the GPU and see the difference between robustness on and off.
class ShaderRobustnessPerf : public DawnPerfTestWithParams<ShaderRobustnessParams> { … };

void ShaderRobustnessPerf::SetUp() { … }

std::string ShaderRobustnessPerf::GetShaderPreamble() { … }

std::string ShaderRobustnessPerf::GetShaderBody() { … }

std::string ShaderRobustnessPerf::GetShader() { … }

void ShaderRobustnessPerf::Step() { … }

TEST_P(ShaderRobustnessPerf, Run) { … }

DAWN_INSTANTIATE_TEST_P(…);

}  // anonymous namespace
}  // namespace dawn