|
161 | 161 | end
|
162 | 162 | @testset "kron" begin
|
163 | 163 | @testset "AbstractVecOrMat{$T}" for T in (Float64, ComplexF64)
|
164 |
| - test_frule(kron, randn(T, 3), randn(T, 3)) |
165 |
| - test_frule(kron, randn(T, 3, 2), randn(T, 3)) |
166 |
| - test_frule(kron, randn(T, 3), randn(T, 3, 4)) |
167 |
| - test_frule(kron, randn(T, 3, 4), randn(T, 2, 2)) |
| 164 | + @testset "frule" begin |
| 165 | + test_frule(kron, randn(T, 3), randn(T, 3)) |
| 166 | + test_frule(kron, randn(T, 3, 2), randn(T, 3)) |
| 167 | + test_frule(kron, randn(T, 3), randn(T, 3, 4)) |
| 168 | + test_frule(kron, randn(T, 3, 4), randn(T, 2, 2)) |
| 169 | + end |
| 170 | + @testset "rrule" begin |
| 171 | + test_rrule(kron, randn(T, 3), randn(T, 3)) |
| 172 | + test_rrule(kron, randn(T, 3, 2), randn(T, 3)) |
| 173 | + test_rrule(kron, randn(T, 3), randn(T, 3, 4)) |
| 174 | + test_rrule(kron, randn(T, 3, 4), randn(T, 2, 2)) |
| 175 | + end |
168 | 176 | end
|
169 | 177 | end
|
170 | 178 | end
|
0 commit comments