diff --git a/cpp/include/cugraph_c/types.h b/cpp/include/cugraph_c/types.h index 3303ca3e28..dddb802b93 100644 --- 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 index 0010e83c3a..384b588c79 100644 --- 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