add additional types (unused)

rapidsai · Oct 25, 2024 · 527c081 · 527c081
1 parent 5b466fa
commit 527c081
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/cpp/include/cugraph_c/types.h b/cpp/include/cugraph_c/types.h
@@ -27,8 +27,14 @@ typedef enum bool_ { FALSE = 0, TRUE = 1 } bool_t;
 typedef int8_t byte_t;
 
 typedef enum data_type_id_ {
-  INT32 = 0,
+  INT8 = 0,
+  INT16,
+  INT32,
   INT64,
+  UINT8,
+  UINT16,
+  UINT32,
+  UINT64,
   FLOAT32,
   FLOAT64,
   SIZE_T,

diff --git a/python/pylibcugraph/pylibcugraph/_cugraph_c/types.pxd b/python/pylibcugraph/pylibcugraph/_cugraph_c/types.pxd
@@ -24,10 +24,17 @@ cdef extern from "cugraph_c/types.h":
         TRUE
 
     ctypedef enum cugraph_data_type_id_t:
+        INT8
+        INT16
         INT32
         INT64
+        UINT8
+        UINT16
+        UINT32
+        UINT64
         FLOAT32
         FLOAT64
         SIZE_T
+        BOOL
 
     ctypedef int8_t byte_t