buffer align

hana-alice · Sep 5, 2023 · 37fa034 · 37fa034
1 parent 5c021f2
commit 37fa034
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/native/cocos/renderer/gfx-wgpu/WGPUBuffer.cpp b/native/cocos/renderer/gfx-wgpu/WGPUBuffer.cpp
@@ -27,13 +27,15 @@
 #include "WGPUDevice.h"
 #include "WGPUObject.h"
 #include "WGPUUtils.h"
+#include <boost/align/align_up.hpp>
 
 namespace cc {
 namespace gfx {
 
 namespace {
 CCWGPUBuffer *dftUniformBuffer = nullptr;
 CCWGPUBuffer *dftStorageBuffer = nullptr;
+static uint32_t BUFFER_ALIGNMENT = 16;
 } // namespace
 
 using namespace emscripten;
@@ -54,7 +56,7 @@ void CCWGPUBuffer::doInit(const BufferInfo &info) {
         _gpuBufferObject->indirectObjs.resize(drawInfoCount);
     }
 
-    _size = ceil(info.size / 16.0) * 16;
+    _size = boost::alignment::align_up(_size, BUFFER_ALIGNMENT);
 
     WGPUBufferDescriptor descriptor = {
         .nextInChain = nullptr,
@@ -113,7 +115,7 @@ void CCWGPUBuffer::doResize(uint32_t size, uint32_t count) {
         _gpuBufferObject->indirectObjs.resize(drawInfoCount);
     }
 
-    _size = ceil(size / 16.0) * 16;
+    _size = boost::alignment::align_up(_size, BUFFER_ALIGNMENT);
 
     WGPUBufferDescriptor descriptor = {
         .nextInChain = nullptr,
@@ -172,7 +174,7 @@ void CCWGPUBuffer::update(const void *buffer, uint32_t size) {
         update(drawInfo, drawInfoCount);
     } else {
         size_t offset = _isBufferView ? _offset : 0;
-        uint32_t alignedSize = ceil(size / 16.0) * 16;
+        uint32_t alignedSize = boost::alignment::align_up(size, BUFFER_ALIGNMENT);
         size_t buffSize = alignedSize;
         wgpuQueueWriteBuffer(CCWGPUDevice::getInstance()->gpuDeviceObject()->wgpuQueue, _gpuBufferObject->wgpuBuffer, offset, buffer, buffSize);
     }