pytorch - 💡(How to fix) Fix `torch.compile` silently returns a result for `torch.linalg.cholesky` on non-positive-definite input

import torch

A = torch.tensor(
    [
        [1.0, 2.0],
        [2.0, 1.0],
    ],
    dtype=torch.float32,
)

def fn(x):
    return torch.linalg.cholesky(x)

try:
    y_eager = fn(A)
    print("eager ok:")
    print(y_eager)
except Exception as e:
    print("eager error:")
    print(type(e).__name__)
    print(e)

try:
    compiled_fn = torch.compile(fn, backend="inductor", fullgraph=True)
    y_compiled = compiled_fn(A)
    print("compiled ok:")
    print(y_compiled)
except Exception as e:
    print("compiled error:")
    print(type(e).__name__)
    print(e)

---

eager error:
_LinAlgError
linalg.cholesky: The factorization could not be completed because the input is not positive-definite (the leading minor of order 2 is not positive-definite).
compiled ok:
tensor([[ 1.0000,  0.0000],
        [ 2.0000, -3.0000]])