Skip to content

Commit 5c4cb8d

Browse files
committed
Fix tests
1 parent 12fc593 commit 5c4cb8d

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
@@ -9355,7 +9355,7 @@ static int iq2_find_best_neighbour(const uint16_t * restrict neighbours, const u
93559355

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

9358-
const int gindex = iq2_data_index(256);
9358+
const int gindex = iq2_data_index(GGML_TYPE_IQ2_XXS);
93599359

93609360
const uint64_t * kgrid_q2xs = iq2_data[gindex].grid;
93619361
const int * kmap_q2xs = iq2_data[gindex].map;
@@ -9528,7 +9528,7 @@ static void quantize_row_iq2_xxs_impl(const float * restrict x, void * restrict
95289528

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

9531-
const int gindex = iq2_data_index(512);
9531+
const int gindex = iq2_data_index(GGML_TYPE_IQ2_XS);
95329532

95339533
const uint64_t * kgrid_q2xs = iq2_data[gindex].grid;
95349534
const int * kmap_q2xs = iq2_data[gindex].map;

ggml.c

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19143,7 +19143,8 @@ size_t ggml_quantize_q8_0(const float * src, void * dst, int n, int k, int64_t *
1914319143
bool ggml_quantize_requires_imatrix(enum ggml_type type) {
1914419144
return
1914519145
type == GGML_TYPE_IQ2_XXS ||
19146-
type == GGML_TYPE_IQ2_XS;
19146+
type == GGML_TYPE_IQ2_XS ||
19147+
type == GGML_TYPE_IQ1_S;
1914719148
}
1914819149

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

0 commit comments

Comments
 (0)