Skip to content

Commit 082ec0e

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

File tree

1 file changed

+15
-7
lines changed

1 file changed

+15
-7
lines changed

test/mooncake.jl

Lines changed: 15 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,10 @@ 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 (
255+
LAPACK_Simple(),
256+
#LAPACK_Expert(), # expensive on CI
257+
)
255258
@testset "eig_full" begin
256259
Mooncake.TestUtils.test_rule(rng, eig_full, A, alg; mode = Mooncake.ReverseMode, output_tangent = dDV, is_primitive = false, atol = atol, rtol = rtol)
257260
test_pullbacks_match(rng, eig_full!, eig_full, A, (D, V), (ΔD2, ΔV), alg)
@@ -340,9 +343,9 @@ MatrixAlgebraKit.copy_input(::typeof(copy_eigh_trunc), A) = MatrixAlgebraKit.cop
340343
Ddiag = diagview(D)
341344
@testset for alg in (
342345
LAPACK_QRIteration(),
343-
LAPACK_DivideAndConquer(),
344-
LAPACK_Bisection(),
345-
LAPACK_MultipleRelativelyRobustRepresentations(),
346+
#LAPACK_DivideAndConquer(),
347+
#LAPACK_Bisection(),
348+
#LAPACK_MultipleRelativelyRobustRepresentations(), # expensive on CI
346349
)
347350
@testset "eigh_full" begin
348351
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 +393,7 @@ end
390393
minmn = min(m, n)
391394
@testset for alg in (
392395
LAPACK_QRIteration(),
393-
LAPACK_DivideAndConquer(),
396+
#LAPACK_DivideAndConquer(), # expensive on CI
394397
)
395398
@testset "svd_compact" begin
396399
ΔU = randn(rng, T, m, minmn)
@@ -490,7 +493,12 @@ end
490493
@testset "size ($m, $n)" for n in (17, m, 23)
491494
atol = rtol = m * n * precision(T)
492495
A = randn(rng, T, m, n)
493-
@testset for alg in PolarViaSVD.((LAPACK_QRIteration(), LAPACK_DivideAndConquer()))
496+
@testset for alg in PolarViaSVD.(
497+
(
498+
LAPACK_QRIteration(),
499+
#LAPACK_DivideAndConquer(), # expensive on CI
500+
)
501+
)
494502
if m >= n
495503
WP = left_polar(A, alg)
496504
Mooncake.TestUtils.test_rule(rng, left_polar, A, alg; mode = Mooncake.ReverseMode, is_primitive = false, atol = atol, rtol = rtol)

0 commit comments

Comments
 (0)