gausplat_renderer::render::gaussian_3d::jit

Function forward

source
pub fn forward<R: JitRuntime, F: FloatElement, I: IntElement, B: BoolElement>(
    input: RenderInput<JitBackend<R, F, I, B>>,
    view: &View,
    options: &Gaussian3dRenderOptions,
) -> Result<RenderOutput<JitBackend<R, F, I, B>>, Error>
Expand description

Render the 3DGS scene (forward).

It computes the colors in RGB space from the 3DGS scene.

The kernels are launched in the following order:

  1. Transform the scene parameters.
  2. Scan the counts of the touched tiles into offsets.
  3. Rank the points by its tile index and depth.
  4. Sort the points by its tile index and depth.
  5. Segment the points by its tile index.
  6. Rasterize the points into the image.