Add .pxd file for Cython utils

rapidsai · Aug 11, 2020 · 920c3b8 · 920c3b8
1 parent 55ce3d2
commit 920c3b8
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/ucp/_libs/utils.pxd b/ucp/_libs/utils.pxd
@@ -0,0 +1,11 @@
+# Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
+# See file LICENSE for terms.
+
+# cython: language_level=3
+
+
+from libc.stdint cimport uintptr_t
+
+
+cpdef uintptr_t get_buffer_data(buffer, bint check_writable=*) except *
+cpdef Py_ssize_t get_buffer_nbytes(buffer, check_min_size, bint cuda_support) except *