From 471ec0b614794d22fc945e60d6546e25ba68bb66 Mon Sep 17 00:00:00 2001 From: Joseph Kleinhenz Date: Sat, 3 Aug 2024 20:54:11 -0700 Subject: [PATCH] comment --- src/beignet/func/_implicit_diff_root_scalar.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/beignet/func/_implicit_diff_root_scalar.py b/src/beignet/func/_implicit_diff_root_scalar.py index 4639900b21..37ac35bbeb 100644 --- a/src/beignet/func/_implicit_diff_root_scalar.py +++ b/src/beignet/func/_implicit_diff_root_scalar.py @@ -42,7 +42,7 @@ def backward(ctx, *grad_outputs): -g * b / A for g, b in zip(grad_outputs, B, strict=True) ) elif A.ndim == 2: - # NOTE even in this case we should always have A diagonal because f is scalar + # NOTE even in this case A is diagonal because f is scalar return tuple( torch.linalg.solve(A, -g * b) for g, b in zip(grad_outputs, B, strict=True)