Просмотр исходного кода

ggml : use `uint8x16_t` return type for `ggml_vqtbl1q_u8` (#5894)

* use uint8x16_t

* Update ggml-quants.c
bobqianic 1 год назад
Родитель
Сommit
e25fb4b18f
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      ggml-quants.c

+ 2 - 2
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]];