Improve trsm stability

compintell · Dec 11, 2024 · dd087fb · dd087fb
1 parent 7aa53b9
commit dd087fb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/rrules/blas.jl b/src/rrules/blas.jl
@@ -869,7 +869,10 @@ function generate_derived_rrule!!_test_cases(rng_ctor, ::Val{:blas})
         ) do (side, ul, tA, dA, M, N, P)
             t = tA == 'N'
             R = side == 'L' ? M : N
-            As = blas_matrices(rng, P, R, R)
+            As = map(blas_matrices(rng, P, R, R)) do A
+                A[diagind(A)] .+= 1
+                return A
+            end
             Bs = blas_matrices(rng, P, M, N)
             return map_prod(As, Bs) do (A, B)
                 (false, :none, nothing, BLAS.trsm!, side, ul, tA, dA, randn(P), A, B)