ENH: add uint8 (#47)

* add the `uint8` type, an 8-bit unsigned integer whose
values exist on the interval `[0, +255]`

* this type is required by the Python array API standard:
https://data-apis.org/array-api/latest/API_specification/data_types.html

Author: tylerjereddy

include/fwd.hpp

@@ -127,6 +127,7 @@ enum KokkosViewDataType {
   Int16,
   Int32,
   Int64,
+  Uint8,
   Uint16,
   Uint32,
   Uint64,

include/traits.hpp

@@ -79,6 +79,7 @@ VIEW_DATA_TYPE(int8_t, Int8, "int8", "signed_char")
 VIEW_DATA_TYPE(int16_t, Int16, "int16", "short")
 VIEW_DATA_TYPE(int32_t, Int32, "int32", "int")
 VIEW_DATA_TYPE(int64_t, Int64, "int64", "long")
+VIEW_DATA_TYPE(uint8_t, Uint8, "uint8", "unsigned_char")
 VIEW_DATA_TYPE(uint16_t, Uint16, "uint16", "unsigned_short")
 VIEW_DATA_TYPE(uint32_t, Uint32, "uint32", "unsigned", "unsigned_int")
 VIEW_DATA_TYPE(uint64_t, Uint64, "uint64", "unsigned_long")

kokkos/__init__.py.in

@@ -151,6 +151,7 @@ try:
         "int16",
         "int32",
         "int64",
+        "uint8",
         "uint16",
         "uint32",
         "uint64",

kokkos/test/test_types.py

@@ -11,6 +11,7 @@
         (kokkos.int16, np.int16),
         (kokkos.int32, np.int32),
         (kokkos.int64, np.int64),
+        (kokkos.uint8, np.uint8),
         (kokkos.uint16, np.uint16),
         (kokkos.uint32, np.uint32),
         (kokkos.uint64, np.uint64),

kokkos/utility.py

@@ -89,6 +89,8 @@ def read_dtype(_dtype):
             return lib.int32
         elif _dtype == np.int64:
             return lib.int64
+        elif _dtype == np.uint8:
+            return lib.uint8
         elif _dtype == np.uint16:
             return lib.uint16
         elif _dtype == np.uint32:

src/variants/CMakeLists.txt

@@ -26,7 +26,7 @@ TARGET_LINK_LIBRARIES(libpykokkos-variants PUBLIC
 
 SET(_types              concrete dynamic)
 SET(_variants           layout memory_trait)
-SET(_data_types         Int8 Int16 Int32 Int64 Uint16 Uint32 Uint64 Float32 Float64)
+SET(_data_types         Int8 Int16 Int32 Int64 Uint8 Uint16 Uint32 Uint64 Float32 Float64)
 
 SET(layout_enums        Right)
 SET(memory_trait_enums  Managed)