Skip to content

Commit 2ee4281

Browse files
committed
Fix tests
1 parent fe9c0af commit 2ee4281

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

ggml-quants.c

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9651,7 +9651,7 @@ static int iq2_find_best_neighbour(const uint16_t * restrict neighbours, const u
96519651

96529652
static void quantize_row_iq2_xxs_impl(const float * restrict x, void * restrict vy, int n, const float * restrict quant_weights) {
96539653

9654-
const int gindex = iq2_data_index(256);
9654+
const int gindex = iq2_data_index(GGML_TYPE_IQ2_XXS);
96559655

96569656
const uint64_t * kgrid_q2xs = iq2_data[gindex].grid;
96579657
const int * kmap_q2xs = iq2_data[gindex].map;
@@ -9824,7 +9824,7 @@ static void quantize_row_iq2_xxs_impl(const float * restrict x, void * restrict
98249824

98259825
static void quantize_row_iq2_xs_impl(const float * restrict x, void * restrict vy, int n, const float * restrict quant_weights) {
98269826

9827-
const int gindex = iq2_data_index(512);
9827+
const int gindex = iq2_data_index(GGML_TYPE_IQ2_XS);
98289828

98299829
const uint64_t * kgrid_q2xs = iq2_data[gindex].grid;
98309830
const int * kmap_q2xs = iq2_data[gindex].map;

ggml.c

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19199,7 +19199,8 @@ size_t ggml_quantize_q8_0(const float * src, void * dst, int n, int k, int64_t *
1919919199
bool ggml_quantize_requires_imatrix(enum ggml_type type) {
1920019200
return
1920119201
type == GGML_TYPE_IQ2_XXS ||
19202-
type == GGML_TYPE_IQ2_XS;
19202+
type == GGML_TYPE_IQ2_XS ||
19203+
type == GGML_TYPE_IQ1_S;
1920319204
}
1920419205

1920519206
size_t ggml_quantize_chunk(enum ggml_type type, const float * src, void * dst, int start,

0 commit comments

Comments
 (0)