ggml : use uint8x16_t return type for ggml_vqtbl1q_u8 (ggml-org#5894

) * use uint8x16_t * Update ggml-quants.c
hazelnutcloud · Mar 10, 2024 · acb4657 · acb4657
1 parent 1f745cd
commit acb4657
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ggml-quants.c b/ggml-quants.c
@@ -464,8 +464,8 @@ inline static int8x16_t ggml_vqtbl1q_s8(int8x16_t a, uint8x16_t b) {
 }
 
 // NOTE: not tested
-inline static int8x16_t ggml_vqtbl1q_u8(uint8x16_t a, uint8x16_t b) {
-    int8x16_t res;
+inline static uint8x16_t ggml_vqtbl1q_u8(uint8x16_t a, uint8x16_t b) {
+    uint8x16_t res;
 
     res[ 0] = a[b[ 0]];
     res[ 1] = a[b[ 1]];