diff --git a/NDTensors/test/NDTensorsTestUtils/device_list.jl b/NDTensors/test/NDTensorsTestUtils/device_list.jl index 56c44d5fc4..de4639bafb 100644 --- a/NDTensors/test/NDTensorsTestUtils/device_list.jl +++ b/NDTensors/test/NDTensorsTestUtils/device_list.jl @@ -23,6 +23,10 @@ if "cutensor" in ARGS || "all" in ARGS Pkg.add("cuTENSOR") using CUDA, cuTENSOR end +if "jlarrays" in ARGS || "all" in ARGS + Pkg.add("JLArrays") + using JLArrays +end function devices_list(test_args) devs = Vector{Function}(undef, 0) @@ -47,5 +51,10 @@ function devices_list(test_args) if "metal" in test_args || "all" in test_args push!(devs, NDTensors.MetalExtensions.mtl) end + + if "jlarrays" in test_args || "all" in test_args + push!(devs, JLArrays.jl) + end + return devs end