Skip to main content

dispatch_tri_solve

Function dispatch_tri_solve 

Source
pub fn dispatch_tri_solve(
    encoder: &mut CommandEncoder,
    registry: &mut KernelRegistry,
    device: &DeviceRef,
    l: &MlxBuffer,
    b: &MlxBuffer,
    x: &MlxBuffer,
    params_buf: &MlxBuffer,
    p: TriSolveParams,
) -> Result<()>
Expand description

Dispatch a lower-triangular unit-diagonal solve.