Skip to content

Commit 445770f

Browse files
authored
Ops remaining basic numerical test (EnzymeAD#320)
1 parent be1b423 commit 445770f

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

test/ops.jl

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,11 @@ end
320320
@test log.(Array(x)) @jit Ops.log(x)
321321
end
322322

323-
@testset "logistic" begin end
323+
@testset "logistic" begin
324+
x = ConcreteRArray([0.0, 1.0, 2.0, 3.0])
325+
l(x) = 1 / (1 + exp(-x))
326+
@test l.(Array(x)) @jit Ops.logistic(x)
327+
end
324328

325329
@testset "maximum" begin
326330
x = ConcreteRArray([false, false, true, true])
@@ -798,4 +802,8 @@ end
798802
@test (; values=[4, 3], indices=[3, 2]) == @jit Ops.top_k(x, 2)
799803
end
800804

801-
@testset "zeta" begin end
805+
@testset "zeta" begin
806+
s = ConcreteRArray([1.0, 2.0, 50.0])
807+
z = ConcreteRArray([1e-8, 0.001, 2.0])
808+
@test SpecialFunctions.zeta.(Array(s), Array(z)) @jit Ops.zeta(s, z)
809+
end

0 commit comments

Comments
 (0)