Skip to content

Commit 540e146

Browse files
committed
Comment out some Mooncake tests to slim down CI runs
1 parent f3de4de commit 540e146

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

test/mooncake.jl

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ make_mooncake_tangent(T::Tuple) = Mooncake.build_tangent(typeof(T), T...)
2222
make_mooncake_fdata(x) = make_mooncake_tangent(x)
2323
make_mooncake_fdata(x::Diagonal) = Mooncake.FData((diag = make_mooncake_tangent(x.diag),))
2424

25-
ETs = (Float64, Float32, ComplexF64, ComplexF32)
25+
ETs = (Float32, ComplexF64,)
2626

2727
# no `alg` argument
2828
function _get_copying_derivative(f_c, rrule, A, ΔA, args, Δargs, ::Nothing, rdata)
@@ -251,7 +251,9 @@ end
251251
dV = make_mooncake_tangent(ΔV)
252252
dDV = Mooncake.build_tangent(typeof((ΔD2, ΔV)), dD, dV)
253253
# compute the dA corresponding to the above dD, dV
254-
@testset for alg in (LAPACK_Simple(), LAPACK_Expert())
254+
@testset for alg in (LAPACK_Simple(),
255+
#LAPACK_Expert(), # expensive on CI
256+
)
255257
@testset "eig_full" begin
256258
Mooncake.TestUtils.test_rule(rng, eig_full, A, alg; mode = Mooncake.ReverseMode, output_tangent = dDV, is_primitive = false, atol = atol, rtol = rtol)
257259
test_pullbacks_match(rng, eig_full!, eig_full, A, (D, V), (ΔD2, ΔV), alg)
@@ -340,9 +342,9 @@ MatrixAlgebraKit.copy_input(::typeof(copy_eigh_trunc), A) = MatrixAlgebraKit.cop
340342
Ddiag = diagview(D)
341343
@testset for alg in (
342344
LAPACK_QRIteration(),
343-
LAPACK_DivideAndConquer(),
344-
LAPACK_Bisection(),
345-
LAPACK_MultipleRelativelyRobustRepresentations(),
345+
#LAPACK_DivideAndConquer(),
346+
#LAPACK_Bisection(),
347+
#LAPACK_MultipleRelativelyRobustRepresentations(), # expensive on CI
346348
)
347349
@testset "eigh_full" begin
348350
Mooncake.TestUtils.test_rule(rng, copy_eigh_full, A, alg; mode = Mooncake.ReverseMode, output_tangent = dDV, is_primitive = false, atol = atol, rtol = rtol)
@@ -390,7 +392,7 @@ end
390392
minmn = min(m, n)
391393
@testset for alg in (
392394
LAPACK_QRIteration(),
393-
LAPACK_DivideAndConquer(),
395+
#LAPACK_DivideAndConquer(), # expensive on CI
394396
)
395397
@testset "svd_compact" begin
396398
ΔU = randn(rng, T, m, minmn)
@@ -490,7 +492,9 @@ end
490492
@testset "size ($m, $n)" for n in (17, m, 23)
491493
atol = rtol = m * n * precision(T)
492494
A = randn(rng, T, m, n)
493-
@testset for alg in PolarViaSVD.((LAPACK_QRIteration(), LAPACK_DivideAndConquer()))
495+
@testset for alg in PolarViaSVD.((LAPACK_QRIteration(),
496+
#LAPACK_DivideAndConquer(), # expensive on CI
497+
))
494498
if m >= n
495499
WP = left_polar(A, alg)
496500
Mooncake.TestUtils.test_rule(rng, left_polar, A, alg; mode = Mooncake.ReverseMode, is_primitive = false, atol = atol, rtol = rtol)

0 commit comments

Comments
 (0)