encounter
/
dawn-cmake
mirror of https://github.com/encounter/dawn-cmake.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make storage buffer length UBO size a multiple of 16 bytes 

Conditional on whether TINT_EXPECTS_UBOS_TO_BE_MULTIPLE_OF_16 is
defined, which it will be in a future Tint CL.

Bug: tint:984
Change-Id: I191cd59fe6947c9113a8fad7dbfbf79903969b27
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/57960
Commit-Queue: Antonio Maiorano <amaiorano@google.com>
Reviewed-by: Austin Eng <enga@chromium.org>
This commit is contained in:
Antonio Maiorano 2021-07-14 15:33:31 +00:00 committed by Dawn LUCI CQ
parent 954e8abe48
commit 58016bfb31
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/dawn_native/metal/CommandBufferMTL.mm
View File

@ -30,6 +30,8 @@
#include "dawn_native/metal/TextureMTL.h"
#include "dawn_native/metal/UtilsMetal.h"
#include <tint/tint.h>
namespace dawn_native { namespace metal {
namespace {
@ -272,7 +274,7 @@ namespace dawn_native { namespace metal {
// The lengths of buffers are stored as 32bit integers because that is the width the
// MSL code generated by SPIRV-Cross expects.
PerStage<std::array<uint32_t, kGenericMetalBufferSlots>> data;
PerStage<std::array<uint32_t, ((kGenericMetalBufferSlots / 4) + 1) * 4>> data;
void Apply(id<MTLRenderCommandEncoder> render,
RenderPipeline* pipeline,
@ -292,6 +294,10 @@ namespace dawn_native { namespace metal {
bufferCount += pipeline->GetVertexBufferCount();
}
#ifdef TINT_EXPECTS_UBOS_TO_BE_MULTIPLE_OF_16
bufferCount = Align(bufferCount, 4);
ASSERT(bufferCount <= data[SingleShaderStage::Vertex].size());
#endif
[render setVertexBytes:data[SingleShaderStage::Vertex].data()
length:sizeof(uint32_t) * bufferCount
atIndex:kBufferLengthBufferSlot];
@ -300,6 +306,10 @@ namespace dawn_native { namespace metal {
if (stagesToApply & wgpu::ShaderStage::Fragment) {
uint32_t bufferCount = ToBackend(pipeline->GetLayout())
->GetBufferBindingCount(SingleShaderStage::Fragment);
#ifdef TINT_EXPECTS_UBOS_TO_BE_MULTIPLE_OF_16
bufferCount = Align(bufferCount, 4);
ASSERT(bufferCount <= data[SingleShaderStage::Fragment].size());
#endif
[render setFragmentBytes:data[SingleShaderStage::Fragment].data()
length:sizeof(uint32_t) * bufferCount
atIndex:kBufferLengthBufferSlot];
@ -320,6 +330,10 @@ namespace dawn_native { namespace metal {
uint32_t bufferCount = ToBackend(pipeline->GetLayout())
->GetBufferBindingCount(SingleShaderStage::Compute);
#ifdef TINT_EXPECTS_UBOS_TO_BE_MULTIPLE_OF_16
bufferCount = Align(bufferCount, 4);
ASSERT(bufferCount <= data[SingleShaderStage::Compute].size());
#endif
[compute setBytes:data[SingleShaderStage::Compute].data()
length:sizeof(uint32_t) * bufferCount
atIndex:kBufferLengthBufferSlot];