From 3ca2e3890be0a3c8b66ac9e2c8ef9c26800c9ced Mon Sep 17 00:00:00 2001
From: Cedric Chevalier <cedric.chevalier@cea.fr>
Date: Wed, 13 Dec 2023 21:18:48 +0100
Subject: [PATCH] More interesting sum.

---
 examples/kokkos/simple_example.cpp | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/examples/kokkos/simple_example.cpp b/examples/kokkos/simple_example.cpp
index ee4aaf0..1d36183 100644
--- a/examples/kokkos/simple_example.cpp
+++ b/examples/kokkos/simple_example.cpp
@@ -41,19 +41,20 @@ int main(int argc, char* argv[]) {
     Sdouble sum = 0;
 
     Kokkos::parallel_reduce(
-      n, KOKKOS_LAMBDA(const int i, Sdouble& lsum) { lsum += i * i; }, sum);
+      n, KOKKOS_LAMBDA(const int i, Sdouble& lsum) { auto val = Sdouble(i+1); lsum += 1/val * 1/val; }, sum);
 
     std::cout <<
-            "Sum of squares of integers from 0 to " << n-1 <<
+            "Sum from 0 to " << n-1 <<
             ", computed in parallel, is " << sum << std::endl;
 
     // Compare to a sequential loop.
     Sdouble seqSum = 0;
     for (int i = 0; i < n; ++i) {
-        seqSum += i * i;
+        Sdouble val = i + 1;
+        seqSum += 1/val * 1/val;
     }
     std::cout <<
-                 "Sum of squares of integers from 0 to " << n-1 <<
+                 "Sum from 0 to " << n-1 <<
                  ", computed in sequential, is " << seqSum << std::endl;
 
     Kokkos::finalize();