@@ -228,7 +228,7 @@ int main(int argc, char * argv[]) {
228
228
if (qfns.quantize_row_q && qfns.dequantize_row_q ) {
229
229
printf (" %s\n " , ggml_type_name (type));
230
230
231
- if (params.op_quantize_row_q_reference ) {
231
+ if (params.op_quantize_row_q_reference && qfns. quantize_row_q_reference ) {
232
232
printf (" quantize_row_q_reference\n " );
233
233
for (size_t size : params.test_sizes ) {
234
234
printf (" %zu values (%.2f MB)\n " , size, 4 *size/(float )(1024 *1024 ));
@@ -242,7 +242,7 @@ int main(int argc, char * argv[]) {
242
242
printf (" \n " );
243
243
}
244
244
245
- if (params.op_quantize_row_q ) {
245
+ if (params.op_quantize_row_q && qfns. quantize_row_q ) {
246
246
printf (" quantize_row_q\n " );
247
247
for (size_t size : params.test_sizes ) {
248
248
printf (" %zu values (%.2f MB)\n " , size, 4 *size/(float )(1024 *1024 ));
@@ -256,7 +256,7 @@ int main(int argc, char * argv[]) {
256
256
printf (" \n " );
257
257
}
258
258
259
- if (params.op_dequantize_row_q ) {
259
+ if (params.op_dequantize_row_q && qfns. dequantize_row_q ) {
260
260
printf (" dequantize_row_q\n " );
261
261
qfns.quantize_row_q (test_data1, test_q1, largest);
262
262
for (size_t size : params.test_sizes ) {
@@ -271,7 +271,7 @@ int main(int argc, char * argv[]) {
271
271
printf (" \n " );
272
272
}
273
273
274
- if (params.op_quantize_row_q_dot ) {
274
+ if (params.op_quantize_row_q_dot && qfns. quantize_row_q_dot ) {
275
275
printf (" quantize_row_q_dot\n " );
276
276
for (size_t size : params.test_sizes ) {
277
277
printf (" %zu values (%.2f MB)\n " , size, 4 *size/(float )(1024 *1024 ));
@@ -285,7 +285,7 @@ int main(int argc, char * argv[]) {
285
285
printf (" \n " );
286
286
}
287
287
288
- if (params.op_vec_dot_q ) {
288
+ if (params.op_vec_dot_q && qfns. vec_dot_q ) {
289
289
printf (" vec_dot_q\n " );
290
290
qfns.quantize_row_q (test_data1, test_q1, largest);
291
291
qfns.quantize_row_q (test_data2, test_q2, largest);
0 commit comments