Skip to main content

define_extension_runtime

Macro define_extension_runtime 

Source
define_extension_runtime!() { /* proc-macro */ }
Expand description

Generate a standard extension module, preparation engine, and prepared operation.

The execute function must have this signature: fn<B: BackendBound + 'static>(&OpType, &[&Tensor], &mut ExtensionExecutionContext<'_, B>).

execute_reads is required. It must have this signature: fn<B: BackendBound + 'static>(&OpType, &[TensorRead<'_>], &mut ExtensionExecutionContext<'_, B>).

session_supported and execute_in_session are optional, but must be supplied together. The former has signature fn<B: BackendBound + 'static>(&OpType) -> bool; the latter has signature fn(&OpType, &mut dyn BackendSession, &mut ExtensionCacheStore, &[TensorRead<'_>]).