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

benchmark : fix result validation in benchmark-q4_0-matmult (#987)

Ivan Komarov 2 лет назад
Родитель
Сommit
c12b14b77f
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      examples/benchmark/benchmark-q4_0-matmult.c

+ 1 - 1
examples/benchmark/benchmark-q4_0-matmult.c

@@ -24,7 +24,7 @@
 
 float tensor_sum_elements(struct ggml_tensor * tensor) {
     float sum = 0;
-    if (tensor->type==6) {
+    if (tensor->type==GGML_TYPE_F32) {
         for (int j = 0; j < tensor->ne[1]; j++) {
             for (int k = 0; k < tensor->ne[0]; k++) {
                 sum +=  ((float *) tensor->data)[j*tensor->ne[0]+k];