Change dimensions to make them recognizable

JuliaDiff · Sep 26, 2023 · 3ccbdfc · 3ccbdfc
1 parent b9b7f10
commit 3ccbdfc
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/test/rulesets/LinearAlgebra/dense.jl b/test/rulesets/LinearAlgebra/dense.jl
@@ -162,16 +162,16 @@
     @testset "kron" begin
         @testset "AbstractVecOrMat{$T}" for T in (Float64, ComplexF64)
             @testset "frule" begin
-                test_frule(kron, randn(T, 3), randn(T, 3))
-                test_frule(kron, randn(T, 3, 2), randn(T, 3))
-                test_frule(kron, randn(T, 3), randn(T, 3, 4))
-                test_frule(kron, randn(T, 3, 4), randn(T, 2, 2))
+                test_frule(kron, randn(T, 2), randn(T, 3))
+                test_frule(kron, randn(T, 2, 3), randn(T, 5))
+                test_frule(kron, randn(T, 2), randn(T, 3, 5))
+                test_frule(kron, randn(T, 2, 3), randn(T, 5, 7))
             end
             @testset "rrule" begin
-                test_rrule(kron, randn(T, 3), randn(T, 3))
-                test_rrule(kron, randn(T, 3, 2), randn(T, 3))
-                test_rrule(kron, randn(T, 3), randn(T, 3, 4))
-                test_rrule(kron, randn(T, 3, 4), randn(T, 2, 2))
+                test_rrule(kron, randn(T, 2), randn(T, 3))
+                test_rrule(kron, randn(T, 2, 3), randn(T, 5))
+                test_rrule(kron, randn(T, 2), randn(T, 3, 5))
+                test_rrule(kron, randn(T, 2, 3), randn(T, 5, 7))
             end
         end
     end