From c443ceea8c85cd747cdce6f3332ec860aa4d4cc7 Mon Sep 17 00:00:00 2001
From: Pierre Kestener <pierre.kestener@cea.fr>
Date: Fri, 1 Dec 2023 10:00:42 +0100
Subject: [PATCH] Update reducer types allowed in parallel reduction.

---
 examples/kokkos/random_sum.py         |  2 +-
 pykokkos/core/translators/bindings.py | 18 +++++++++++++++++-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/examples/kokkos/random_sum.py b/examples/kokkos/random_sum.py
index 3da72b07..201cb52a 100644
--- a/examples/kokkos/random_sum.py
+++ b/examples/kokkos/random_sum.py
@@ -7,7 +7,7 @@
 class RandomSum:
     def __init__(self, n):
         self.N: int = n
-        self.total: int = 0
+        self.total: pk.int32 = 0
         self.a: pk.View1D[pk.int32] = pk.View([n], pk.int32)
 
         for i in range(self.N):
diff --git a/pykokkos/core/translators/bindings.py b/pykokkos/core/translators/bindings.py
index cf9b21ca..da5c4c0f 100644
--- a/pykokkos/core/translators/bindings.py
+++ b/pykokkos/core/translators/bindings.py
@@ -232,7 +232,7 @@ def get_return_type(operation: str, workunit: cppast.MethodDecl) -> str:
     :param operation: the type of the operation (for, reduce, scan, or workload)
     :param workunit: the workunit for which the binding is being generated
     :returns: the return type as a string
-    """ 
+    """
 
     acc_decl: Optional[cppast.ParmVarDecl] = None
     if operation == "reduce":
@@ -575,6 +575,22 @@ def bind_main_single(
         if "pk.Acc" in element:
             if "pk.int64" in element:
                 acc_type = "int64_t"
+            elif "pk.int32" in element:
+                acc_type = "int32_t"
+            elif "pk.int16" in element:
+                acc_type = "int16_t"
+            elif "pk.int8" in element:
+                acc_type = "int8_t"
+            elif "pk.uint64" in element:
+                acc_type = "uint64_t"
+            elif "pk.uint32" in element:
+                acc_type = "uint32_t"
+            elif "pk.uint16" in element:
+                acc_type = "uint16_t"
+            elif "pk.uint8" in element:
+                acc_type = "uint8_t"
+            elif "pk.float" in element:
+                acc_type = "float"
             elif "pk.double" in element:
                 acc_type = "double"