Struct ComputeFn

pub struct ComputeFn { /* private fields */ }

An instance of a compute function holding the implementation vtable and a set of registered compute kernels.

Implementations

impl ComputeFn

pub fn new(id: ArcRef<str>, vtable: ArcRef<dyn ComputeFnVTable>) -> Self

Create a new compute function from the given ComputeFnVTable.

pub fn id(&self) -> &ArcRef<str>

Returns the string identifier of the compute function.

pub fn register_kernel(&self, kernel: ArcRef<dyn Kernel>)

Register a kernel for the compute function.

pub fn invoke(&self, args: &InvocationArgs<'_>) -> VortexResult<Output>

Invokes the compute function with the given arguments.

pub fn return_dtype(&self, args: &InvocationArgs<'_>) -> VortexResult<DType>

Compute the return type of the function given the input arguments.

pub fn return_len(&self, args: &InvocationArgs<'_>) -> VortexResult<usize>

Compute the return length of the function given the input arguments.

pub fn is_elementwise(&self) -> bool

Returns whether the compute function is elementwise, i.e. the output is the same shape as

pub fn kernels(&self) -> Vec<ArcRef<dyn Kernel>>

Returns the compute function’s kernels.