From de9fba0d393840015663c601706edf08f8e19d5d Mon Sep 17 00:00:00 2001
From: Jared Van Bortel <jared@nomic.ai>
Date: Thu, 25 Jan 2024 15:22:11 -0500
Subject: [PATCH] kompute : fix basic f16 get_rows, 28 -> 26 failures

---
 kompute-shaders/op_getrows_f16.comp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kompute-shaders/op_getrows_f16.comp b/kompute-shaders/op_getrows_f16.comp
index 34acbcd70..48c936108 100644
--- a/kompute-shaders/op_getrows_f16.comp
+++ b/kompute-shaders/op_getrows_f16.comp
@@ -27,5 +27,5 @@ void main() {
     const uint i = gl_WorkGroupID.x;
     const int r = inB[i + pcs.inBOff];
 
-    dequantize_row_f16(r*pcs.nb01/2/*bytes for float16*/ + pcs.inAOff, i*pcs.nb1 + pcs.outOff, pcs.ne00);
+    dequantize_row_f16(r*pcs.nb01/2/*bytes for float16*/ + pcs.inAOff, i*pcs.nb1/4 + pcs.outOff, pcs.ne00);
 }