Przeglądaj źródła

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

Georgi Gerganov 1 rok temu
rodzic
commit
1debe72737
1 zmienionych plików z 1 dodań i 1 usunięć
  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);