Sat Mar 2 10:37:45 PM EST 2024

diff --git a/src/gpu_mmul.cl b/src/gpu_mmul.cl
index 1f8e4f87690d3df3e1e406430052b26bcbd3d9f7..31cada3f02b39273d4268a5540b55c702122b99f 100644 (file)
--- a/src/gpu_mmul.cl
+++ b/src/gpu_mmul.cl
@@ -8,14 +8,17 @@ __kernel void gpu_mmul
     const int shared
 )
 {
-    const int colsA = shared;
-    const int rowsB = shared;
+    int colsA = shared;
+    int rowsB = shared;
+    int rowsR = rowsA;
+    int colsR = colsB;
     int row = get_global_id(0);
     int col = get_global_id(1);
     float sum = 0;
-    for (int i = 0; i < rowsA; i++)
+
+    for (int i = 0; i < colsR; i++)
     {
         sum += ptrA[row * colsA + i] * ptrB[i * colsB + col];
     }
-    ptrR[row * colsB + col] = sum;
+    ptrR[row * colsR + col] = sum;
 }
\ No newline at end of file