warp.tile_upper_solve#

warp.tile_upper_solve(
U: Tile[Float, tuple[int, int]],
z: Tile[Float, tuple[int]],
) → Tile[Float, tuple[int]]#

  • Kernel

Solve for x in Ux = z, where U is an upper triangular matrix.

This performs general back substitution for upper triangular systems.

Note that computing the adjoint is not yet supported.

Supported datatypes are:

  • float32

  • float64

Parameters:

  • U – A square, non-singular, upper triangular matrix

  • z – A 1D or 2D tile with compatible shape

Returns:

A tile of the same shape as z such that Ux = z.