Sun Mar 3 11:03:08 PM EST 2024

diff --git a/src/gpu/gpu.c b/src/gpu/gpu.c
index 2519c5290eeaa26358d1fdb3ba9132ba239145ec..38d3dfe9d2dcf590d7337b65459c72fe3ff07ef9 100644 (file)
--- a/src/gpu/gpu.c
+++ b/src/gpu/gpu.c
@@ -259,7 +259,7 @@ void GPU_mmul(float* ptrR, float* ptrA, float* ptrB, size_t rowsA, size_t colsB,
        }
 }
 
-void GPU_knk(float* ptrR, size_t rowsR, size_t colsR, float* ptrA, size_t rowsA, size_t colsA, float* ptrB, size_t rowsB, size_t colsB)
+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;