diff --git a/Project.toml b/Project.toml index 26fd5f59..6eb7fb42 100644 --- a/Project.toml +++ b/Project.toml @@ -19,6 +19,7 @@ Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c" [weakdeps] FastBroadcast = "7034ab61-46d4-4ed7-9d0f-46aef9175898" ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210" +KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c" Measurements = "eff96d63-e80a-5855-80a2-b1b0885c5ab7" MonteCarloMeasurements = "0987c9cc-fe09-11e8-30f0-b96dd679fdca" ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267" @@ -30,6 +31,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" [extensions] RecursiveArrayToolsFastBroadcastExt = "FastBroadcast" RecursiveArrayToolsForwardDiffExt = "ForwardDiff" +RecursiveArrayToolsKernelAbstractionsExt = "KernelAbstractions" RecursiveArrayToolsMeasurementsExt = "Measurements" RecursiveArrayToolsMonteCarloMeasurementsExt = "MonteCarloMeasurements" RecursiveArrayToolsReverseDiffExt = ["ReverseDiff", "Zygote"] @@ -47,6 +49,7 @@ FastBroadcast = "0.2.8, 0.3" ForwardDiff = "0.10.19, 1" GPUArraysCore = "0.1.1, 0.2" IteratorInterfaceExtensions = "1" +KernelAbstractions = "0.9.36" LinearAlgebra = "1.10" Measurements = "2.3" MonteCarloMeasurements = "1.1" @@ -74,6 +77,7 @@ julia = "1.10" Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595" FastBroadcast = "7034ab61-46d4-4ed7-9d0f-46aef9175898" ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210" +KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c" Measurements = "eff96d63-e80a-5855-80a2-b1b0885c5ab7" MonteCarloMeasurements = "0987c9cc-fe09-11e8-30f0-b96dd679fdca" NLsolve = "2774e3e8-f4cf-5e23-947b-6d7e65073b56" @@ -90,4 +94,4 @@ Unitful = "1986cc42-f94f-5a68-af5c-568840ba703d" Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" [targets] -test = ["Aqua", "FastBroadcast", "ForwardDiff", "Measurements", "NLsolve", "Pkg", "Random", "SafeTestsets", "SciMLBase", "SparseArrays", "StaticArrays", "StructArrays", "Test", "Unitful", "Zygote"] +test = ["Aqua", "FastBroadcast", "ForwardDiff", "KernelAbstractions", "Measurements", "NLsolve", "Pkg", "Random", "SafeTestsets", "SciMLBase", "SparseArrays", "StaticArrays", "StructArrays", "Test", "Unitful", "Zygote"] diff --git a/ext/RecursiveArrayToolsKernelAbstractions.jl b/ext/RecursiveArrayToolsKernelAbstractionsExt.jl similarity index 88% rename from ext/RecursiveArrayToolsKernelAbstractions.jl rename to ext/RecursiveArrayToolsKernelAbstractionsExt.jl index 94f79884..f4e5f9e6 100644 --- a/ext/RecursiveArrayToolsKernelAbstractions.jl +++ b/ext/RecursiveArrayToolsKernelAbstractionsExt.jl @@ -1,4 +1,4 @@ -module RecursiveArrayToolsKernelAbstractions +module RecursiveArrayToolsKernelAbstractionsExt import RecursiveArrayTools: VectorOfArray import KernelAbstractions diff --git a/test/utils_test.jl b/test/utils_test.jl index 5ccda25a..63f78238 100644 --- a/test/utils_test.jl +++ b/test/utils_test.jl @@ -138,3 +138,9 @@ end @test u1.u[1] isa SVector @test u1.u[2] isa SVector end + +import KernelAbstractions: get_backend +@testset "KernelAbstractions" begin + v = VectorOfArray([randn(2) for i in 1:10]) + @test get_backend(v) === get_backend(parent(v)[1]) +end