Use simpler implementation for reference

ginkgo-project · Oct 19, 2023 · d152e73 · d152e73
1 parent 7b15bdb
commit d152e73
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/reference/matrix/dense_kernels.cpp b/reference/matrix/dense_kernels.cpp
@@ -408,11 +408,10 @@ void compute_mean(std::shared_ptr<const ReferenceExecutor> exec,
     }
 
     for (size_type i = 0; i < x->get_size()[0]; ++i) {
-        const ValueType_nc alpha = static_cast<ValueType_nc>(i) / (i + 1);
-        const ValueType_nc beta = static_cast<ValueType_nc>(1) / (i + 1);
         for (size_type j = 0; j < x->get_size()[1]; ++j) {
-            result->at(0, j) = alpha * result->at(0, j) + beta * x->at(i, j);
+            result->at(0, i) += x->at(i, j);
         }
+        result->at(0, i) /= static_cast<ValueType_nc>(x->get_size()[1]);
     }
 }