Skip to main content

trtx_create_gpu_allocator

Function trtx_create_gpu_allocator 

Source
pub unsafe extern "C" fn trtx_create_gpu_allocator(
    rust_impl: *mut c_void,
    allocateAsync: unsafe extern "system" fn(this: *const c_void, size: u64, alignment: u64, flags: u32, cuda_stream: *mut c_void) -> *mut c_void,
    reallocate: unsafe extern "system" fn(this: *const c_void, memory: *mut c_void, alignment: u64, new_size: u64) -> *mut c_void,
    deallocateAsync: unsafe extern "system" fn(this: *const c_void, memory: *mut c_void, cuda_stream: *mut c_void) -> bool,
) -> *mut IGpuAllocator