fixup! mean computation

ginkgo-project · Jun 27, 2024 · 1d813ee · 1d813ee
1 parent cfa695f
commit 1d813ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/common/unified/matrix/dense_kernels.template.cpp b/common/unified/matrix/dense_kernels.template.cpp
@@ -263,7 +263,7 @@ void compute_mean(std::shared_ptr<const DefaultExecutor> exec,
             return x(i, j) * inv_total_size;
         },
         GKO_KERNEL_REDUCE_SUM(ValueType), result->get_values(), x->get_size(),
-        tmp, x, ValueType_nc{1.} / x->get_size()[0]);
+        tmp, x, ValueType_nc{1.} / std::max(1ul, x->get_size()[0]));
 }