Skip to content

Commit 80c8ced

Browse files
committed
Don't test enzyme on windows
1 parent 87750c4 commit 80c8ced

File tree

3 files changed

+15
-59
lines changed

3 files changed

+15
-59
lines changed

.github/workflows/CI.yml

+1
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ jobs:
2828
os:
2929
- ubuntu-latest
3030
- macos-latest
31+
- windows-latest
3132
steps:
3233
- uses: actions/checkout@v4
3334
- uses: julia-actions/setup-julia@v1

.github/workflows/CI_Windows.yml

-55
This file was deleted.

test/core/rootfind.jl

+14-4
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,16 @@
11
using BenchmarkTools, LinearSolve, NonlinearSolve, StaticArrays, Random, LinearAlgebra,
22
Test, ForwardDiff, Zygote, Enzyme, SparseDiffTools, DiffEqBase
33

4+
function __autosparseenzyme()
5+
@static if Sys.iswindows()
6+
@warn "Enzyme on Windows stalls. Using AutoSparseFiniteDiff instead till \
7+
https://github.com/EnzymeAD/Enzyme.jl/issues/1236 is resolved."
8+
return AutoSparseFiniteDiff()
9+
else
10+
return AutoSparseEnzyme()
11+
end
12+
end
13+
414
_nameof(x) = applicable(nameof, x) ? nameof(x) : _nameof(typeof(x))
515

616
quadratic_f(u, p) = u .* u .- p
@@ -95,7 +105,7 @@ const TERMINATION_CONDITIONS = [
95105
@test nlprob_iterator_interface(quadratic_f!, p, Val(true)) sqrt.(p)
96106

97107
@testset "ADType: $(autodiff) u0: $(_nameof(u0))" for autodiff in (AutoSparseForwardDiff(),
98-
AutoSparseFiniteDiff(), AutoZygote(), AutoSparseZygote(), AutoSparseEnzyme()), u0 in (1.0, [1.0, 1.0])
108+
AutoSparseFiniteDiff(), AutoZygote(), AutoSparseZygote(), __autosparseenzyme()), u0 in (1.0, [1.0, 1.0])
99109
probN = NonlinearProblem(quadratic_f, u0, 2.0)
100110
@test all(solve(probN, NewtonRaphson(; autodiff)).u .≈ sqrt(2.0))
101111
end
@@ -175,7 +185,7 @@ end
175185
@test nlprob_iterator_interface(quadratic_f!, p, Val(true)) sqrt.(p)
176186

177187
@testset "ADType: $(autodiff) u0: $(_nameof(u0)) radius_update_scheme: $(radius_update_scheme)" for autodiff in (AutoSparseForwardDiff(),
178-
AutoSparseFiniteDiff(), AutoZygote(), AutoSparseZygote(), AutoSparseEnzyme()), u0 in (1.0, [1.0, 1.0]),
188+
AutoSparseFiniteDiff(), AutoZygote(), AutoSparseZygote(), __autosparseenzyme()), u0 in (1.0, [1.0, 1.0]),
179189
radius_update_scheme in radius_update_schemes
180190

181191
probN = NonlinearProblem(quadratic_f, u0, 2.0)
@@ -279,7 +289,7 @@ end
279289
end
280290

281291
@testset "ADType: $(autodiff) u0: $(_nameof(u0))" for autodiff in (AutoSparseForwardDiff(),
282-
AutoSparseFiniteDiff(), AutoZygote(), AutoSparseZygote(), AutoSparseEnzyme()), u0 in (1.0, [1.0, 1.0])
292+
AutoSparseFiniteDiff(), AutoZygote(), AutoSparseZygote(), __autosparseenzyme()), u0 in (1.0, [1.0, 1.0])
283293
probN = NonlinearProblem(quadratic_f, u0, 2.0)
284294
@test all(solve(probN, LevenbergMarquardt(; autodiff); abstol = 1e-9,
285295
reltol = 1e-9).u .≈ sqrt(2.0))
@@ -494,7 +504,7 @@ end
494504
@test nlprob_iterator_interface(quadratic_f!, p, Val(true)) sqrt.(p)
495505

496506
@testset "ADType: $(autodiff) u0: $(_nameof(u0))" for autodiff in (AutoSparseForwardDiff(),
497-
AutoSparseFiniteDiff(), AutoZygote(), AutoSparseZygote(), AutoSparseEnzyme()), u0 in (1.0, [1.0, 1.0])
507+
AutoSparseFiniteDiff(), AutoZygote(), AutoSparseZygote(), __autosparseenzyme()), u0 in (1.0, [1.0, 1.0])
498508
probN = NonlinearProblem(quadratic_f, u0, 2.0)
499509
@test all(solve(probN, PseudoTransient(; alpha_initial = 10.0, autodiff)).u .≈
500510
sqrt(2.0))

0 commit comments

Comments
 (0)