From: miha-q <>
Date: Sun, 3 Mar 2024 03:44:34 +0000 (-0500)
Subject: Sat Mar  2 10:44:34 PM EST 2024
X-Git-Url: http://www.foleosoft.com/?a=commitdiff_plain;h=7b190f308647d48561b8dc0d5afd9f3e85e7257a;p=QAnsel.git

Sat Mar  2 10:44:34 PM EST 2024
---

diff --git a/src/gpu_mmul.cl b/src/gpu_mmul.cl
index 94b43d0..7eb1b3f 100644
--- a/src/gpu_mmul.cl
+++ b/src/gpu_mmul.cl
@@ -12,13 +12,18 @@ __kernel void gpu_mmul
     const int rowsB = shared;
     const int rowsR = rowsA;
     const int colsR = colsB;
-    const int row = get_global_id(0);
-    const int col = get_global_id(1);
+    const int rowR = get_global_id(0);
+    const int colR = get_global_id(1);
+    int posA, posB;
     float sum = 0;
 
-    for (int i = 0; i < colsR; i++)
+    const int posR = colR + rowR * colsR;
+
+    for (int i = 0; i < shared; i++)
     {
-        sum += ptrA[row * colsA + i] * ptrB[i * colsB + col];
+        int posA = i + rowR * colsA;
+        int posB = colR + i * colsB;
+        sum += ptrA[posA] * ptrB[posB];
     }
     ptrR[row * colsR + col] = sum;
 }
\ No newline at end of file