test adapt

Author: Pietro Vertechi

Project.toml

@@ -12,10 +12,11 @@ Tables = "1"
 julia = "1"
 
 [extras]
+GPUArrays = "0c68f7d7-f131-5f86-a1c3-88cf8149b2d7"
 OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
 PooledArrays = "2dfb63ee-cc39-5dd5-95bd-886bf059d720"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 WeakRefStrings = "ea10d353-3f73-51f8-a26c-33c1cb351aa5"
 
 [targets]
-test = ["Test", "OffsetArrays", "PooledArrays", "WeakRefStrings"]
+test = ["Test", "GPUArrays", "OffsetArrays", "PooledArrays", "WeakRefStrings"]

src/compatibility.jl

@@ -0,0 +1,24 @@
+# GPU storage
+import Adapt
+Adapt.adapt_structure(to, s::StructArray) = replace_storage(x->Adapt.adapt(to, x), s)
+
+# Table interface
+import Tables
+
+Tables.istable(::Type{<:StructVector}) = true
+Tables.rowaccess(::Type{<:StructVector}) = true
+Tables.columnaccess(::Type{<:StructVector}) = true
+
+Tables.rows(s::StructVector) = s
+Tables.columns(s::StructVector) = fieldarrays(s)
+
+Tables.schema(s::StructVector) = Tables.Schema(staticschema(eltype(s)))
+
+# refarray interface
+import DataAPI: refarray, refvalue
+
+refarray(s::StructArray) = StructArray(map(refarray, fieldarrays(s)))
+
+function refvalue(s::StructArray{T}, v::Tup) where {T}
+    createinstance(T, map(refvalue, fieldarrays(s), v)...)
+end

test/runtests.jl

@@ -3,6 +3,8 @@ using StructArrays: staticschema, iscompatible, _promote_typejoin, append!!
 using OffsetArrays: OffsetArray
 import Tables, PooledArrays, WeakRefStrings
 using DataAPI: refarray, refvalue
+using Adapt: adapt
+import GPUArrays
 using Test
 
 @testset "index" begin
@@ -700,3 +702,13 @@ end
         @test vcat(dest, StructVector(makeitr())) == append!!(copy(dest), makeitr())
     end
 end
+
+@testset "adapt" begin
+    s = StructArray(a = 1:10, b = StructArray(c = 1:10, d = 1:10))
+    t = adapt(Array, s)
+    @test propertynames(t) == (:a, :b)
+    @test s == t
+    @test t.a isa Array
+    @test t.b.c isa Array
+    @test t.b.d isa Array
+end