Sun Mar 3 11:22:24 PM EST 2024

diff --git a/src/gpu/gpu.c b/src/gpu/gpu.c
index e50d8c68706d576f62070da92df9e25d9492f3e3..9a9630696272884aad3193b488d0eb067bf70e45 100644 (file)
--- a/src/gpu/gpu.c
+++ b/src/gpu/gpu.c
@@ -154,9 +154,9 @@ void GPU_mmul(float* ptrR, float* ptrA, float* ptrB, int rowsA, int colsB, int s
 void GPU_knk(float* ptrR, int rowsR, int colsR, float* ptrA, int rowsA, int colsA, float* ptrB, int rowsB, int colsB)
 {
        //Create buffers
-       size_t sizeA = rowsA * colsA;
-       size_t sizeB = rowsB * colsB;
-       size_t sizeR = rowsR * colsR;
+       size_t sizeA = rowsA * colsA * 2;
+       size_t sizeB = rowsB * colsB * 2;
+       size_t sizeR = rowsR * colsR * 2;
        cl_int err;
        cl_mem memA = clCreateBuffer(GPU_context, CL_MEM_READ_ONLY, sizeof(float) * sizeA, NULL, &err);
        gpuerr(clCreateBuffer);