@@ -9,37 +9,37 @@ JLArrays.allowscalar(false)
9
9
idx = 2.0 : 0.17 : 19.0
10
10
jlidx = jl (collect (idx))
11
11
@test itp .(idx) == collect (jlitp .(idx)) == collect (jlitp .(jlidx))
12
- @test gradient .(Ref (itp), idx) ==
13
- collect (gradient .(Ref (jlitp), idx)) ==
14
- collect (gradient .(Ref (jlitp), jlidx))
12
+ @test Interpolations . gradient .(Ref (itp), idx) ==
13
+ collect (Interpolations . gradient .(Ref (jlitp), idx)) ==
14
+ collect (Interpolations . gradient .(Ref (jlitp), jlidx))
15
15
16
16
sitp = scale (itp, A_x)
17
17
jlsitp = jl (sitp)
18
18
idx = 1.0 : 0.4 : 39.0
19
19
jlidx = jl (collect (idx))
20
20
@test sitp .(idx) == collect (jlsitp .(idx)) == collect (jlsitp .(jlidx))
21
- @test gradient .(Ref (sitp), idx) ==
22
- collect (gradient .(Ref (jlsitp), idx)) ==
23
- collect (gradient .(Ref (jlsitp), jlidx))
21
+ @test Interpolations . gradient .(Ref (sitp), idx) ==
22
+ collect (Interpolations . gradient .(Ref (jlsitp), idx)) ==
23
+ collect (Interpolations . gradient .(Ref (jlsitp), jlidx))
24
24
25
25
26
26
esitp = extrapolate (sitp, Flat ())
27
27
jlesitp = jl (esitp)
28
28
idx = - 1.0 : 0.84 : 41.0
29
29
jlidx = jl (collect (idx))
30
30
@test esitp .(idx) == collect (jlesitp .(idx)) == collect (jlesitp .(jlidx))
31
- @test gradient .(Ref (esitp), idx) ==
32
- collect (gradient .(Ref (jlesitp), idx)) ==
33
- collect (gradient .(Ref (jlesitp), jlidx))
31
+ @test Interpolations . gradient .(Ref (esitp), idx) ==
32
+ collect (Interpolations . gradient .(Ref (jlesitp), idx)) ==
33
+ collect (Interpolations . gradient .(Ref (jlesitp), jlidx))
34
34
35
35
esitp = extrapolate (sitp, 0.0 )
36
36
jlesitp = jl (esitp)
37
37
idx = - 1.0 : 0.84 : 41.0
38
38
jlidx = jl (collect (idx))
39
39
@test esitp .(idx) == collect (jlesitp .(idx)) == collect (jlesitp .(jlidx))
40
- @test gradient .(Ref (esitp), idx) ==
41
- collect (gradient .(Ref (jlesitp), idx)) ==
42
- collect (gradient .(Ref (jlesitp), jlidx))
40
+ @test Interpolations . gradient .(Ref (esitp), idx) ==
41
+ collect (Interpolations . gradient .(Ref (jlesitp), idx)) ==
42
+ collect (Interpolations . gradient .(Ref (jlesitp), jlidx))
43
43
end
44
44
45
45
@testset " 2d GPU Interpolation" begin
50
50
idx = 2.0 : 0.17 : 19.0
51
51
jlidx = jl (collect (idx))
52
52
@test itp .(idx, idx' ) == collect (jlitp .(idx, idx' )) == collect (jlitp .(jlidx, jlidx' ))
53
- @test gradient .(Ref (itp), idx, idx' ) ==
54
- collect (gradient .(Ref (jlitp), idx, idx' )) ==
55
- collect (gradient .(Ref (jlitp), jlidx, jlidx' ))
56
- @test hessian .(Ref (itp), idx, idx' ) ==
57
- collect (hessian .(Ref (jlitp), idx, idx' )) ==
58
- collect (hessian .(Ref (jlitp), jlidx, jlidx' ))
53
+ @test Interpolations . gradient .(Ref (itp), idx, idx' ) ==
54
+ collect (Interpolations . gradient .(Ref (jlitp), idx, idx' )) ==
55
+ collect (Interpolations . gradient .(Ref (jlitp), jlidx, jlidx' ))
56
+ @test Interpolations . hessian .(Ref (itp), idx, idx' ) ==
57
+ collect (Interpolations . hessian .(Ref (jlitp), idx, idx' )) ==
58
+ collect (Interpolations . hessian .(Ref (jlitp), jlidx, jlidx' ))
59
59
60
60
sitp = scale (itp, A_x, A_x)
61
61
jlsitp = jl (sitp)
62
62
idx = 1.0 : 0.4 : 39.0
63
63
jlidx = jl (collect (idx))
64
64
@test sitp .(idx, idx' ) == collect (jlsitp .(idx, idx' )) == collect (jlsitp .(jlidx, jlidx' ))
65
- @test gradient .(Ref (sitp), idx, idx' ) ==
66
- collect (gradient .(Ref (jlsitp), idx, idx' )) ==
67
- collect (gradient .(Ref (jlsitp), jlidx, jlidx' ))
68
- @test hessian .(Ref (sitp), idx, idx' ) ==
69
- collect (hessian .(Ref (jlsitp), idx, idx' )) ==
70
- collect (hessian .(Ref (jlsitp), jlidx, jlidx' ))
65
+ @test Interpolations . gradient .(Ref (sitp), idx, idx' ) ==
66
+ collect (Interpolations . gradient .(Ref (jlsitp), idx, idx' )) ==
67
+ collect (Interpolations . gradient .(Ref (jlsitp), jlidx, jlidx' ))
68
+ @test Interpolations . hessian .(Ref (sitp), idx, idx' ) ==
69
+ collect (Interpolations . hessian .(Ref (jlsitp), idx, idx' )) ==
70
+ collect (Interpolations . hessian .(Ref (jlsitp), jlidx, jlidx' ))
71
71
72
72
esitp = extrapolate (sitp, Flat ())
73
73
jlesitp = jl (esitp)
74
74
idx = - 1.0 : 0.84 : 41.0
75
75
jlidx = jl (collect (idx))
76
76
@test esitp .(idx, idx' ) == collect (jlesitp .(idx, idx' )) == collect (jlesitp .(jlidx, jlidx' ))
77
- # gradient for `extrapolation` is currently broken under CUDA
78
- @test gradient .(Ref (esitp), idx, idx' ) ==
79
- collect (gradient .(Ref (jlesitp), idx, idx' )) ==
80
- collect (gradient .(Ref (jlesitp), jlidx, jlidx' ))
77
+ # Interpolations. gradient for `extrapolation` is currently broken under CUDA
78
+ @test Interpolations . gradient .(Ref (esitp), idx, idx' ) ==
79
+ collect (Interpolations . gradient .(Ref (jlesitp), idx, idx' )) ==
80
+ collect (Interpolations . gradient .(Ref (jlesitp), jlidx, jlidx' ))
81
81
82
82
esitp = extrapolate (sitp, 0.0 )
83
83
jlesitp = jl (esitp)
84
84
idx = - 1.0 : 0.84 : 41.0
85
85
jlidx = jl (collect (idx))
86
86
@test esitp .(idx, idx' ) == collect (jlesitp .(idx, idx' )) == collect (jlesitp .(jlidx, jlidx' ))
87
- # gradient for `extrapolation` is currently broken under CUDA
88
- @test gradient .(Ref (esitp), idx, idx' ) ==
89
- collect (gradient .(Ref (jlesitp), idx, idx' )) ==
90
- collect (gradient .(Ref (jlesitp), jlidx, jlidx' ))
87
+ # Interpolations. gradient for `extrapolation` is currently broken under CUDA
88
+ @test Interpolations . gradient .(Ref (esitp), idx, idx' ) ==
89
+ collect (Interpolations . gradient .(Ref (jlesitp), idx, idx' )) ==
90
+ collect (Interpolations . gradient .(Ref (jlesitp), jlidx, jlidx' ))
91
91
end
92
92
93
93
@testset " Lanczos on gpu" begin
0 commit comments