stream.pxd 201 Bytes
Newer Older
root's avatar
root committed
1
2
3
4
5
6
from libc.stdint cimport intptr_t

cdef intptr_t get_current_stream_ptr()
cdef set_current_stream_ptr(intptr_t ptr, int device_id=*)
cpdef intptr_t get_default_stream_ptr()
cdef bint is_ptds_enabled()