test: more fixes

Author: avik-pal

test/integration/linear_algebra.jl

@@ -383,8 +383,10 @@ end
             B = rand(T, 4, 3)
             B_ra = Reactant.to_rarray(B)
 
-            @test @jit(solve_with_lu(A_ra, b_ra)) ≈ solve_with_lu(A, b)
-            @test @jit(solve_with_lu(A_ra, B_ra)) ≈ solve_with_lu(A, B)
+            @test @jit(solve_with_lu(A_ra, b_ra)) ≈ solve_with_lu(A, b) atol = 1e-4 rtol =
+                1e-2
+            @test @jit(solve_with_lu(A_ra, B_ra)) ≈ solve_with_lu(A, B) atol = 1e-4 rtol =
+                1e-2
         end
     end
 
@@ -401,8 +403,10 @@ end
             B = rand(T, 4, 5, 3, 2)
             B_ra = Reactant.to_rarray(B)
 
-            @test @jit(solve_with_lu(A_ra, b_ra)) ≈ solve_with_lu_batched(A, b)
-            @test @jit(solve_with_lu(A_ra, B_ra)) ≈ solve_with_lu_batched(A, B)
+            @test @jit(solve_with_lu(A_ra, b_ra)) ≈ solve_with_lu_batched(A, b) atol = 1e-4 rtol =
+                1e-2
+            @test @jit(solve_with_lu(A_ra, B_ra)) ≈ solve_with_lu_batched(A, B) atol = 1e-4 rtol =
+                1e-2
         end
     end
 
@@ -412,6 +416,7 @@ end
         A_ra = Reactant.to_rarray(A)
         B_ra = Reactant.to_rarray(B)
 
-        @test @jit(solve_with_lu(A_ra, B_ra)) ≈ solve_with_lu_batched(A, B)
+        @test @jit(solve_with_lu(A_ra, B_ra)) ≈ solve_with_lu_batched(A, B) atol = 1e-4 rtol =
+            1e-2
     end
 end

test/integration/special_functions.jl

@@ -9,7 +9,8 @@ macro ≈(a, b)
 end
 
 @testset "gamma" begin
-    @test SpecialFunctions.gamma(0.5) ≈ @jit(SpecialFunctions.gamma(ConcreteRNumber(0.5)))
+    @test SpecialFunctions.gamma(0.5) ≈ @jit(SpecialFunctions.gamma(ConcreteRNumber(0.5))) atol =
+        1e-5 rtol = 1e-3
     @test SpecialFunctions.gamma(Int32(2)) ≈
         @jit(SpecialFunctions.gamma(ConcreteRNumber(Int32(2)))) atol = 1e-5 rtol = 1e-3
 end