From dbd2836d3fe147ceeed6c1811450dc18ccef13c5 Mon Sep 17 00:00:00 2001
From: "William S. Moses" <gh@wsmoses.com>
Date: Mon, 23 Dec 2024 10:50:13 -0500
Subject: [PATCH] Add jlarrays test

---
 test/Project.toml    |  1 +
 test/ext/jlarrays.jl | 10 ++++++++++
 test/runtests.jl     |  1 +
 3 files changed, 12 insertions(+)
 create mode 100644 test/ext/jlarrays.jl

diff --git a/test/Project.toml b/test/Project.toml
index 818e0ac708..fbc6d754fe 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -13,6 +13,7 @@ LLVM = "929cbde3-209d-540e-8aea-75f648917ca0"
 LLVM_jll = "86de99a1-58d6-5da7-8064-bd56ce2e322c"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LogExpFunctions = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
+JLArrays = "27aeb0d3-9eb9-45fb-866b-73c2ecf80fcb"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
diff --git a/test/ext/jlarrays.jl b/test/ext/jlarrays.jl
new file mode 100644
index 0000000000..f1997d1003
--- /dev/null
+++ b/test/ext/jlarrays.jl
@@ -0,0 +1,10 @@
+using Enzyme, Test, JLArrays
+
+function jlres(x)
+    2 * x
+end
+
+@testset "JLArrays" begin
+    Enzyme.jacobian(Forward, jlres, JLArray([3.0, 5.0]))
+    Enzyme.jacobian(Reverse, jlres, JLArray([3.0, 5.0]))
+end
diff --git a/test/runtests.jl b/test/runtests.jl
index c9cde02c28..7f6e4700f0 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -3778,5 +3778,6 @@ include("ext/logexpfunctions.jl")
     include("ext/bfloat16s.jl")
 end
 
+include("ext/jlarrays.jl")
 include("ext/sparsearrays.jl")
 include("ext/staticarrays.jl")