thek0tyara/llama-cpp-turboquant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

Fixed bug by passing size of bytes instead of size of array

Browse source
This commit is contained in:
Alejandro Saucedo 2020-08-12 21:35:20 +01:00
parent 8d04b51cbf
commit 03c697ba11
1 changed files with 2 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/main.cpp
View file

@ -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);