Skip to content

Commit b235cef

Browse files
committed
update enzyme
1 parent cb2f486 commit b235cef

9 files changed

Lines changed: 14 additions & 14 deletions

File tree

test/enzyme/eig.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1414

1515
m = 19
1616
for T in (BLASFloats..., GenericFloats...)
17-
TestSuite.seed_rng!(123)
17+
TestSuite.seed_rng!(1234)
1818
if !is_buildkite
1919
TestSuite.test_enzyme_eig(T, (m, m); atol = m * m * TestSuite.precision(T), rtol = m * m * TestSuite.precision(T))
2020
end

test/enzyme/eigh.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1414

1515
m = 19
1616
for T in (BLASFloats..., GenericFloats...)
17-
TestSuite.seed_rng!(123)
17+
TestSuite.seed_rng!(1234)
1818
if !is_buildkite
1919
TestSuite.test_enzyme_eigh(T, (m, m); atol = m * m * TestSuite.precision(T), rtol = m * m * TestSuite.precision(T))
2020
end

test/enzyme/lq.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ using Test
33
using LinearAlgebra: Diagonal
44
using CUDA, AMDGPU
55

6-
BLASFloats = (Float32, ComplexF64) # full suite is too expensive on CI
6+
BLASFloats = (Float64, ComplexF64) # full suite is too expensive on CI
77
GenericFloats = ()
88
@isdefined(TestSuite) || include("../testsuite/TestSuite.jl")
99
using .TestSuite
@@ -12,7 +12,7 @@ is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1212

1313
m = 19
1414
for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
15-
TestSuite.seed_rng!(123)
15+
TestSuite.seed_rng!(1234)
1616
if !is_buildkite
1717
TestSuite.test_enzyme_lq(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
1818
end

test/enzyme/orthnull.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ using Test
33
using LinearAlgebra: Diagonal
44
using CUDA, AMDGPU
55

6-
BLASFloats = (Float32, ComplexF64) # full suite is too expensive on CI
6+
BLASFloats = (Float64, ComplexF64) # full suite is too expensive on CI
77
GenericFloats = ()
88
@isdefined(TestSuite) || include("../testsuite/TestSuite.jl")
99
using .TestSuite
@@ -12,7 +12,7 @@ is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1212

1313
m = 19
1414
for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
15-
TestSuite.seed_rng!(123)
15+
TestSuite.seed_rng!(1234)
1616
if !is_buildkite
1717
TestSuite.test_enzyme_orthnull(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
1818
end

test/enzyme/polar.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ using Test
33
using LinearAlgebra: Diagonal
44
using CUDA, AMDGPU
55

6-
BLASFloats = (Float32, ComplexF64) # full suite is too expensive on CI
6+
BLASFloats = (Float64, ComplexF64) # full suite is too expensive on CI
77
GenericFloats = ()
88
@isdefined(TestSuite) || include("../testsuite/TestSuite.jl")
99
using .TestSuite
@@ -12,7 +12,7 @@ is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1212

1313
m = 19
1414
for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
15-
TestSuite.seed_rng!(123)
15+
TestSuite.seed_rng!(1234)
1616
if !is_buildkite
1717
TestSuite.test_enzyme_polar(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
1818
end

test/enzyme/projections.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ using Test
33
using LinearAlgebra: Diagonal
44
using CUDA, AMDGPU
55

6-
BLASFloats = (Float32, ComplexF64) # full suite is too expensive on CI
6+
BLASFloats = (Float64, ComplexF64) # full suite is too expensive on CI
77
GenericFloats = ()
88
@isdefined(TestSuite) || include("../testsuite/TestSuite.jl")
99
using .TestSuite
@@ -12,7 +12,7 @@ is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1212

1313
m = 19
1414
for T in (BLASFloats..., GenericFloats...)
15-
TestSuite.seed_rng!(123)
15+
TestSuite.seed_rng!(1234)
1616
atol = rtol = m * m * TestSuite.precision(T)
1717
if !is_buildkite
1818
TestSuite.test_enzyme_projections(T, (m, m); atol, rtol)

test/enzyme/qr.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ using Test
33
using LinearAlgebra: Diagonal
44
using CUDA, AMDGPU
55

6-
BLASFloats = (Float32, ComplexF64) # full suite is too expensive on CI
6+
BLASFloats = (Float64, ComplexF64) # full suite is too expensive on CI
77
GenericFloats = ()
88
@isdefined(TestSuite) || include("../testsuite/TestSuite.jl")
99
using .TestSuite
@@ -12,7 +12,7 @@ is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1212

1313
m = 19
1414
for T in (BLASFloats..., GenericFloats...), n in (17, m, 23)
15-
TestSuite.seed_rng!(123)
15+
TestSuite.seed_rng!(1234)
1616
if !is_buildkite
1717
TestSuite.test_enzyme_qr(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
1818
end

test/testsuite/enzyme/lq.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ function test_enzyme_lq_compact_rank_deficient(
3838
r = min(m, n) - 5
3939
A = instantiate_matrix(T, (m, r)) * instantiate_matrix(T, (r, n))
4040
alg = MatrixAlgebraKit.select_algorithm(lq_compact, A)
41-
LQ, ΔLQ = ad_lq_rank_deficient_compact_setup(A)
41+
LQ, ΔLQ = ad_lq_compact_setup(A)
4242
test_reverse(lq_compact, RT, (A, TA), (alg, Const); atol, rtol, output_tangent = ΔLQ, fdm)
4343
test_reverse(call_and_zero!, RT, (lq_compact!, Const), (A, TA), (alg, Const); atol, rtol, output_tangent = ΔLQ, fdm)
4444
end

test/testsuite/enzyme/qr.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ function test_enzyme_qr_compact_rank_deficient(
3838
r = min(m, n) - 5
3939
A = instantiate_matrix(T, (m, r)) * instantiate_matrix(T, (r, n))
4040
alg = MatrixAlgebraKit.select_algorithm(qr_compact, A)
41-
QR, ΔQR = ad_qr_rank_deficient_compact_setup(A)
41+
QR, ΔQR = ad_qr_compact_setup(A)
4242
test_reverse(qr_compact, RT, (A, TA), (alg, Const); atol, rtol, output_tangent = ΔQR, fdm)
4343
test_reverse(call_and_zero!, RT, (qr_compact!, Const), (A, TA), (alg, Const); atol, rtol, output_tangent = ΔQR, fdm)
4444
end

0 commit comments

Comments
 (0)