🚀 Feature

cholesky raises a RuntimeError if the matrix is not (numerically) positive-definite. This makes it impossible to use it with TorchScript (for loop over increasing regularization of try-except with cholesky).

Similar to lu, return info to indicate which of the batches matrices were successfully decomposed and which weren't.

Motivation

Use cholesky with TorchScript.

Pitch

Return lu-like info instead of raising a RuntimeError.

cc @vishwakftw @jianyuh @nikitaved @pearu @mruberry @heitorschueroff