From 03c697ba11c46fe64e8c6e1a12773d78d293fac5 Mon Sep 17 00:00:00 2001
From: Alejandro Saucedo <axsauze@gmail.com>
Date: Wed, 12 Aug 2020 21:35:20 +0100
Subject: [PATCH] Fixed bug by passing size of bytes instead of size of array

---
 src/main.cpp | 11 ++---------
 1 file changed, 2 insertions(+), 9 deletions(-)

diff --git a/src/main.cpp b/src/main.cpp
index a3da11089..e1fd86660 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -106,14 +106,6 @@ class VulkanCompute
 
         *aMemory = this->mDevice.allocateMemory(memoryAllocateInfo);
 
-        if (data != nullptr) {
-            vk::DeviceSize offset = 0;
-            void* mapped = this->mDevice.mapMemory(
-                    *aMemory, offset, aSize, {});
-            memcpy(mapped, data, aSize);
-            this->mDevice.unmapMemory(*aMemory);
-        }
-
         this->mDevice.bindBufferMemory(*aBuffer, *aMemory, 0);
     }
 
@@ -292,7 +284,8 @@ class VulkanCompute
         */
         {
             void* mapped = this->mDevice.mapMemory(
-                    hostMemory, 0, VK_WHOLE_SIZE, {});
+                    hostMemory, 0, bufferSize, {});
+            memcpy(mapped, computeInput.data(), bufferSize);
             vk::MappedMemoryRange mappedRange(hostMemory, 0, VK_WHOLE_SIZE);
             this->mDevice.flushMappedMemoryRanges(1, &mappedRange);
             this->mDevice.unmapMemory(hostMemory);