Skip to main content

cholesky_solve_only_gpu

gam_gpu::solver

Function cholesky_solve_only_gpu 

Source 
pub fn cholesky_solve_only_gpu(
    hessian: ArrayView2<'_, f64>,
    rhs: ArrayView2<'_, f64>,
) -> Result<Array2<f64>, String>
Expand description

Solution-only mixed-precision solve: like cholesky_solve_gpu but skips the redundant fp64 POTRF when the fp32 + refinement path succeeds, since the caller does not consume the log-determinant. This is the path that delivers the full mixed-precision speedup (expensive O(p³) factor stays fp32) for the PIRLS Newton direction solve, where the logdet is discarded. The solution is full fp64 accuracy via iterative refinement.