mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-12-14 15:46:28 +00:00
Support buffer lazy initialization before CopyTextureToBuffer
BUG=dawn:414 TEST=dawn_end2end_tests Change-Id: I5bdd6333029170d47ea240388e7b7d80750ae5d9 Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/25643 Reviewed-by: Corentin Wallez <cwallez@chromium.org> Reviewed-by: Bryan Bernhart <bryan.bernhart@intel.com> Reviewed-by: Austin Eng <enga@chromium.org> Commit-Queue: Jiawei Shao <jiawei.shao@intel.com>
This commit is contained in:
committed by
Commit Bot service account
parent
361a0d8334
commit
c11a19145f
@@ -28,6 +28,19 @@
|
||||
} \
|
||||
} while (0)
|
||||
|
||||
namespace {
|
||||
|
||||
struct BufferZeroInitInCopyT2BSpec {
|
||||
wgpu::Extent3D textureSize;
|
||||
uint64_t bufferOffset;
|
||||
uint64_t extraBytes;
|
||||
uint32_t bytesPerRow;
|
||||
uint32_t rowsPerImage;
|
||||
uint32_t lazyClearCount;
|
||||
};
|
||||
|
||||
} // anonymous namespace
|
||||
|
||||
class BufferZeroInitTest : public DawnTest {
|
||||
public:
|
||||
wgpu::Buffer CreateBuffer(uint64_t size,
|
||||
@@ -60,8 +73,9 @@ class BufferZeroInitTest : public DawnTest {
|
||||
}
|
||||
}
|
||||
|
||||
wgpu::Texture CreateAndInitialize2DTexture(const wgpu::Extent3D& size,
|
||||
wgpu::TextureFormat format) {
|
||||
wgpu::Texture CreateAndInitializeTexture(const wgpu::Extent3D& size,
|
||||
wgpu::TextureFormat format,
|
||||
wgpu::Color color = {0.f, 0.f, 0.f, 0.f}) {
|
||||
wgpu::TextureDescriptor descriptor;
|
||||
descriptor.size = size;
|
||||
descriptor.format = format;
|
||||
@@ -70,14 +84,73 @@ class BufferZeroInitTest : public DawnTest {
|
||||
wgpu::Texture texture = device.CreateTexture(&descriptor);
|
||||
|
||||
wgpu::CommandEncoder encoder = device.CreateCommandEncoder();
|
||||
utils::ComboRenderPassDescriptor renderPassDescriptor({texture.CreateView()});
|
||||
wgpu::RenderPassEncoder renderPass = encoder.BeginRenderPass(&renderPassDescriptor);
|
||||
renderPass.EndPass();
|
||||
|
||||
for (uint32_t arrayLayer = 0; arrayLayer < size.depth; ++arrayLayer) {
|
||||
wgpu::TextureViewDescriptor viewDescriptor;
|
||||
viewDescriptor.format = format;
|
||||
viewDescriptor.dimension = wgpu::TextureViewDimension::e2D;
|
||||
viewDescriptor.baseArrayLayer = arrayLayer;
|
||||
viewDescriptor.arrayLayerCount = 1u;
|
||||
|
||||
utils::ComboRenderPassDescriptor renderPassDescriptor(
|
||||
{texture.CreateView(&viewDescriptor)});
|
||||
renderPassDescriptor.cColorAttachments[0].clearColor = color;
|
||||
wgpu::RenderPassEncoder renderPass = encoder.BeginRenderPass(&renderPassDescriptor);
|
||||
renderPass.EndPass();
|
||||
}
|
||||
|
||||
wgpu::CommandBuffer commandBuffer = encoder.Finish();
|
||||
queue.Submit(1, &commandBuffer);
|
||||
|
||||
return texture;
|
||||
}
|
||||
|
||||
void TestBufferZeroInitInCopyTextureToBuffer(const BufferZeroInitInCopyT2BSpec& spec) {
|
||||
constexpr wgpu::TextureFormat kTextureFormat = wgpu::TextureFormat::R32Float;
|
||||
ASSERT(utils::GetTexelBlockSizeInBytes(kTextureFormat) * spec.textureSize.width %
|
||||
kTextureBytesPerRowAlignment ==
|
||||
0);
|
||||
|
||||
constexpr wgpu::Color kClearColor = {0.5f, 0.5f, 0.5f, 0.5f};
|
||||
wgpu::Texture texture =
|
||||
CreateAndInitializeTexture(spec.textureSize, kTextureFormat, kClearColor);
|
||||
|
||||
const wgpu::TextureCopyView textureCopyView =
|
||||
utils::CreateTextureCopyView(texture, 0, {0, 0, 0});
|
||||
|
||||
const uint64_t bufferSize = spec.bufferOffset + spec.extraBytes +
|
||||
utils::RequiredBytesInCopy(spec.bytesPerRow, spec.rowsPerImage,
|
||||
spec.textureSize, kTextureFormat);
|
||||
wgpu::BufferDescriptor bufferDescriptor;
|
||||
bufferDescriptor.size = bufferSize;
|
||||
bufferDescriptor.usage = wgpu::BufferUsage::CopySrc | wgpu::BufferUsage::CopyDst;
|
||||
wgpu::Buffer buffer = device.CreateBuffer(&bufferDescriptor);
|
||||
const wgpu::BufferCopyView bufferCopyView = utils::CreateBufferCopyView(
|
||||
buffer, spec.bufferOffset, spec.bytesPerRow, spec.rowsPerImage);
|
||||
|
||||
wgpu::CommandEncoder encoder = device.CreateCommandEncoder();
|
||||
encoder.CopyTextureToBuffer(&textureCopyView, &bufferCopyView, &spec.textureSize);
|
||||
wgpu::CommandBuffer commandBuffer = encoder.Finish();
|
||||
EXPECT_LAZY_CLEAR(spec.lazyClearCount, queue.Submit(1, &commandBuffer));
|
||||
|
||||
const uint64_t expectedValueCount = bufferSize / sizeof(float);
|
||||
std::vector<float> expectedValues(expectedValueCount, 0.f);
|
||||
|
||||
for (uint32_t slice = 0; slice < spec.textureSize.depth; ++slice) {
|
||||
const uint64_t baseOffsetBytesPerSlice =
|
||||
spec.bufferOffset + spec.bytesPerRow * spec.rowsPerImage * slice;
|
||||
for (uint32_t y = 0; y < spec.textureSize.height; ++y) {
|
||||
const uint64_t baseOffsetBytesPerRow =
|
||||
baseOffsetBytesPerSlice + spec.bytesPerRow * y;
|
||||
const uint64_t baseOffsetFloatCountPerRow = baseOffsetBytesPerRow / sizeof(float);
|
||||
for (uint32_t x = 0; x < spec.textureSize.width; ++x) {
|
||||
expectedValues[baseOffsetFloatCountPerRow + x] = 0.5f;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
EXPECT_BUFFER_FLOAT_RANGE_EQ(expectedValues.data(), buffer, 0, expectedValues.size());
|
||||
}
|
||||
};
|
||||
|
||||
// Test that calling writeBuffer to overwrite the entire buffer doesn't need to lazily initialize
|
||||
@@ -405,7 +478,7 @@ TEST_P(BufferZeroInitTest, CopyBufferToTexture) {
|
||||
|
||||
constexpr wgpu::TextureFormat kTextureFormat = wgpu::TextureFormat::R32Uint;
|
||||
|
||||
wgpu::Texture texture = CreateAndInitialize2DTexture(kTextureSize, kTextureFormat);
|
||||
wgpu::Texture texture = CreateAndInitializeTexture(kTextureSize, kTextureFormat);
|
||||
const wgpu::TextureCopyView textureCopyView =
|
||||
utils::CreateTextureCopyView(texture, 0, {0, 0, 0});
|
||||
|
||||
@@ -461,6 +534,70 @@ TEST_P(BufferZeroInitTest, CopyBufferToTexture) {
|
||||
}
|
||||
}
|
||||
|
||||
// Test that the code path of CopyTextureToBuffer clears the destination buffer correctly when it is
|
||||
// the first use of the buffer and the texture is a 2D non-array texture.
|
||||
TEST_P(BufferZeroInitTest, Copy2DTextureToBuffer) {
|
||||
constexpr wgpu::Extent3D kTextureSize = {64u, 8u, 1u};
|
||||
|
||||
// bytesPerRow == texelBlockSizeInBytes * copySize.width && bytesPerRow * copySize.height ==
|
||||
// buffer.size
|
||||
{
|
||||
TestBufferZeroInitInCopyTextureToBuffer(
|
||||
{kTextureSize, 0u, 0u, kTextureBytesPerRowAlignment, kTextureSize.height, 0u});
|
||||
}
|
||||
|
||||
// bytesPerRow > texelBlockSizeInBytes * copySize.width
|
||||
{
|
||||
constexpr uint64_t kBytesPerRow = kTextureBytesPerRowAlignment * 2;
|
||||
TestBufferZeroInitInCopyTextureToBuffer(
|
||||
{kTextureSize, 0u, 0u, kBytesPerRow, kTextureSize.height, 1u});
|
||||
}
|
||||
|
||||
// bufferOffset > 0
|
||||
{
|
||||
constexpr uint64_t kBufferOffset = 16u;
|
||||
TestBufferZeroInitInCopyTextureToBuffer({kTextureSize, kBufferOffset, 0u,
|
||||
kTextureBytesPerRowAlignment, kTextureSize.height,
|
||||
1u});
|
||||
}
|
||||
|
||||
// bytesPerRow * copySize.height < buffer.size
|
||||
{
|
||||
constexpr uint64_t kExtraBufferSize = 16u;
|
||||
TestBufferZeroInitInCopyTextureToBuffer({kTextureSize, 0u, kExtraBufferSize,
|
||||
kTextureBytesPerRowAlignment, kTextureSize.height,
|
||||
1u});
|
||||
}
|
||||
}
|
||||
|
||||
// Test that the code path of CopyTextureToBuffer clears the destination buffer correctly when it is
|
||||
// the first use of the buffer and the texture is a 2D array texture.
|
||||
TEST_P(BufferZeroInitTest, Copy2DArrayTextureToBuffer) {
|
||||
constexpr wgpu::Extent3D kTextureSize = {64u, 4u, 3u};
|
||||
|
||||
// bytesPerRow == texelBlockSizeInBytes * copySize.width && rowsPerImage == copySize.height &&
|
||||
// bytesPerRow * (rowsPerImage * (copySize.depth - 1) + copySize.height) == buffer.size
|
||||
{
|
||||
TestBufferZeroInitInCopyTextureToBuffer(
|
||||
{kTextureSize, 0u, 0u, kTextureBytesPerRowAlignment, kTextureSize.height, 0u});
|
||||
}
|
||||
|
||||
// rowsPerImage > copySize.height
|
||||
{
|
||||
constexpr uint64_t kRowsPerImage = kTextureSize.height + 1u;
|
||||
TestBufferZeroInitInCopyTextureToBuffer(
|
||||
{kTextureSize, 0u, 0u, kTextureBytesPerRowAlignment, kRowsPerImage, 1u});
|
||||
}
|
||||
|
||||
// bytesPerRow * rowsPerImage * copySize.depth < buffer.size
|
||||
{
|
||||
constexpr uint64_t kExtraBufferSize = 16u;
|
||||
TestBufferZeroInitInCopyTextureToBuffer({kTextureSize, 0u, kExtraBufferSize,
|
||||
kTextureBytesPerRowAlignment, kTextureSize.height,
|
||||
1u});
|
||||
}
|
||||
}
|
||||
|
||||
DAWN_INSTANTIATE_TEST(BufferZeroInitTest,
|
||||
D3D12Backend({"nonzero_clear_resources_on_creation_for_testing",
|
||||
"lazy_clear_buffer_on_first_use"}),
|
||||
|
||||
Reference in New Issue
Block a user