unthunk input to Tridiagonal_pullback

JuliaDiff · May 20, 2024 · 938623b · 938623b
1 parent dfbd363
commit 938623b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/rulesets/LinearAlgebra/structured.jl b/src/rulesets/LinearAlgebra/structured.jl
@@ -275,8 +275,9 @@ end
 
 function rrule(::Type{Tridiagonal}, dl, d, du)
     y = Tridiagonal(dl, d, du)
-    @views function ∇Tridiagonal(∂y)
+    @views function Tridiagonal_pullback(ȳ)
+        ∂y = unthunk(ȳ)
         return (NoTangent(), diag(∂y, -1), diag(∂y), diag(∂y, 1))
     end
-    return y, ∇Tridiagonal
+    return y, Tridiagonal_pullback
 end