Function init

external/ggml/examples/python/ggml/utils.py:8–16

Initialize a ggml context, which will be freed automatically when the pointer is garbage collected.

(mem_size: int, mem_buffer: ffi.CData = ffi.NULL, no_alloc: bool = False)

import numpy as np

def init(mem_size: int, mem_buffer: ffi.CData = ffi.NULL, no_alloc: bool = False) -> ffi.CData:
    """
      Initialize a ggml context, which will be freed automatically when the pointer is garbage collected.
    """
    params = ffi.new('struct ggml_init_params*')
    params.mem_size = mem_size
    params.mem_buffer = mem_buffer
    params.no_alloc = no_alloc
    return ffi.gc(lib.ggml_init(params[0]), lib.ggml_free)
 
TensorLike = Union[ffi.CData, np.ndarray]