Skip to content

Commit 3459acb

Browse files
Update structured.jl
1 parent 675feca commit 3459acb

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

test/rulesets/LinearAlgebra/structured.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,4 +161,9 @@
161161
end
162162
end
163163
end
164+
165+
@testset "Tridiagonal" begin
166+
res, pb = rrule(Tridiagonal, [1, 4], [2, 3, 4], [5, 3])
167+
@test pb(10*res) == (NoTangent(), [10, 40], [20, 30, 40], [50, 30])
168+
end
164169
end

0 commit comments

Comments
 (0)