from libc.stdint cimport intptr_t


cdef class CPointer:
    cdef void* ptr


cdef class Function:

    cdef:
        public Module module
        public intptr_t ptr

    cpdef linear_launch(self, size_t size, args, size_t shared_mem=*,
                        size_t block_max_size=*, stream=*,
                        bint enable_cooperative_groups=*)


cdef class Module:

    cdef:
        public intptr_t ptr
        readonly dict mapping

    cpdef load_file(self, filename)
    cpdef load(self, bytes cubin)
    cpdef get_global_var(self, name)
    cpdef get_function(self, name)
    cpdef _set_mapping(self, dict mapping)


cdef class LinkState:

    cdef:
        public intptr_t ptr

    cpdef add_ptr_data(self, bytes data, unicode name)
    cpdef add_ptr_file(self, unicode path)
    cpdef bytes complete(self)
