Update structured.jl

JuliaDiff · Nov 22, 2023 · 3459acb · 3459acb
1 parent 675feca
commit 3459acb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/rulesets/LinearAlgebra/structured.jl b/test/rulesets/LinearAlgebra/structured.jl
@@ -161,4 +161,9 @@
             end
         end
     end
+
+    @testset "Tridiagonal" begin
+        res, pb = rrule(Tridiagonal, [1, 4], [2, 3, 4], [5, 3])
+        @test pb(10*res) == (NoTangent(), [10, 40], [20, 30, 40], [50, 30])
+    end
 end