Sun Mar 3 11:24:54 PM EST 2024

diff --git a/src/gpu/gpu.c b/src/gpu/gpu.c
index 9a9630696272884aad3193b488d0eb067bf70e45..e9c612cce3ea2246f4049a3e1d50e07c57dda779 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 * 2;
-       size_t sizeB = rowsB * colsB * 2;
-       size_t sizeR = rowsR * colsR * 2;
+       size_t sizeA = (rowsA * 2) * (colsA * 2);
+       size_t sizeB = (rowsB * 2) * (colsB * 2);
+       size_t sizeR = (rowsR * 2) * (colsR * 2);
        cl_int err;
        cl_mem memA = clCreateBuffer(GPU_context, CL_MEM_READ_ONLY, sizeof(float) * sizeA, NULL, &err);
        gpuerr(clCreateBuffer);