fix: gpu allocator issue

PilotLightTech · Jan 28, 2025 · 3145314 · 3145314
1 parent 0328ddc
commit 3145314
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/extensions/pl_gpu_allocators_ext.c b/extensions/pl_gpu_allocators_ext.c
@@ -479,7 +479,7 @@ pl_allocate_buddy(
 {
     plDeviceAllocatorData* ptData = (plDeviceAllocatorData*)ptInst;
 
-    if(ulAlignment > 0)
+    if(ulAlignment > 0 && ulSize < PL_DEVICE_BUDDY_BLOCK_SIZE)
         ulSize = ulSize + (ulAlignment - 1);
 
     const uint32_t uLevel = pl__get_buddy_level(ulSize);
@@ -628,7 +628,7 @@ pl_allocate_staging_uncached_buddy(
 {
     plDeviceAllocatorData* ptData = (plDeviceAllocatorData*)ptInst;
 
-    if(ulAlignment > 0)
+    if(ulAlignment > 0 && ulSize < PL_DEVICE_BUDDY_BLOCK_SIZE)
         ulSize = ulSize + (ulAlignment - 1);
 
     const uint32_t uLevel = pl__get_buddy_level(ulSize);