blob: 91566cab601b74543de672f121bb85a75a7d99fb [file] [log] [blame]
// Copyright 2019 The Dawn Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <vector>
#include "dawn/tests/perf_tests/DawnPerfTest.h"
#include "dawn/utils/WGPUHelpers.h"
namespace dawn {
namespace {
constexpr unsigned int kNumIterations = 50;
enum class UploadMethod {
WriteBuffer,
MappedAtCreation,
};
// Perf delta exists between ranges [0, 1MB] vs [1MB, MAX_SIZE).
// These are sample buffer sizes within each range.
enum class UploadSize {
BufferSize_1KB = 1 * 1024,
BufferSize_64KB = 64 * 1024,
BufferSize_1MB = 1 * 1024 * 1024,
BufferSize_4MB = 4 * 1024 * 1024,
BufferSize_16MB = 16 * 1024 * 1024,
};
struct BufferUploadParams : AdapterTestParam {
BufferUploadParams(const AdapterTestParam& param,
UploadMethod uploadMethod,
UploadSize uploadSize)
: AdapterTestParam(param), uploadMethod(uploadMethod), uploadSize(uploadSize) {}
UploadMethod uploadMethod;
UploadSize uploadSize;
};
std::ostream& operator<<(std::ostream& ostream, const BufferUploadParams& param) {
ostream << static_cast<const AdapterTestParam&>(param);
switch (param.uploadMethod) {
case UploadMethod::WriteBuffer:
ostream << "_WriteBuffer";
break;
case UploadMethod::MappedAtCreation:
ostream << "_MappedAtCreation";
break;
}
switch (param.uploadSize) {
case UploadSize::BufferSize_1KB:
ostream << "_BufferSize_1KB";
break;
case UploadSize::BufferSize_64KB:
ostream << "_BufferSize_64KB";
break;
case UploadSize::BufferSize_1MB:
ostream << "_BufferSize_1MB";
break;
case UploadSize::BufferSize_4MB:
ostream << "_BufferSize_4MB";
break;
case UploadSize::BufferSize_16MB:
ostream << "_BufferSize_16MB";
break;
}
return ostream;
}
// Test uploading |kBufferSize| bytes of data |kNumIterations| times.
class BufferUploadPerf : public DawnPerfTestWithParams<BufferUploadParams> {
public:
BufferUploadPerf()
: DawnPerfTestWithParams(kNumIterations, 1),
data(static_cast<size_t>(GetParam().uploadSize)) {}
~BufferUploadPerf() override = default;
void SetUp() override;
private:
void Step() override;
wgpu::Buffer dst;
std::vector<uint8_t> data;
};
void BufferUploadPerf::SetUp() {
DawnPerfTestWithParams<BufferUploadParams>::SetUp();
wgpu::BufferDescriptor desc = {};
desc.size = data.size();
desc.usage = wgpu::BufferUsage::CopyDst;
dst = device.CreateBuffer(&desc);
}
void BufferUploadPerf::Step() {
switch (GetParam().uploadMethod) {
case UploadMethod::WriteBuffer: {
for (unsigned int i = 0; i < kNumIterations; ++i) {
queue.WriteBuffer(dst, 0, data.data(), data.size());
}
// Make sure all WriteBuffer's are flushed.
queue.Submit(0, nullptr);
break;
}
case UploadMethod::MappedAtCreation: {
wgpu::BufferDescriptor desc = {};
desc.size = data.size();
desc.usage = wgpu::BufferUsage::CopySrc | wgpu::BufferUsage::MapWrite;
desc.mappedAtCreation = true;
wgpu::CommandEncoder encoder = device.CreateCommandEncoder();
for (unsigned int i = 0; i < kNumIterations; ++i) {
wgpu::Buffer buffer = device.CreateBuffer(&desc);
memcpy(buffer.GetMappedRange(0, data.size()), data.data(), data.size());
buffer.Unmap();
encoder.CopyBufferToBuffer(buffer, 0, dst, 0, data.size());
}
wgpu::CommandBuffer commands = encoder.Finish();
queue.Submit(1, &commands);
break;
}
}
}
TEST_P(BufferUploadPerf, Run) {
RunTest();
}
DAWN_INSTANTIATE_TEST_P(BufferUploadPerf,
{D3D12Backend(), MetalBackend(), OpenGLBackend(), VulkanBackend()},
{UploadMethod::WriteBuffer, UploadMethod::MappedAtCreation},
{UploadSize::BufferSize_1KB, UploadSize::BufferSize_64KB,
UploadSize::BufferSize_1MB, UploadSize::BufferSize_4MB,
UploadSize::BufferSize_16MB});
} // anonymous namespace
} // namespace dawn