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

ggml : silence UB sanitizer error during iq2_xxs quantization (#0)

Georgi Gerganov 1 год назад
Родитель
Сommit
1debe72737
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      ggml-quants.c

+ 1 - 1
ggml-quants.c

@@ -12144,7 +12144,7 @@ static void quantize_row_iq2_xxs_impl(const float * restrict x, void * restrict
                     printf("\n");
                     GGML_ASSERT(false);
                 }
-                q2[2*ib+0] |= (grid_index << 8*k);
+                q2[2*ib+0] |= ((uint32_t) grid_index << 8*k);
                 q2[2*ib+1] |= (block_signs[k] << 7*k);
             }
             GGML_ASSERT(scale >= 0);