more PtrArrays

JuliaSIMD · May 27, 2022 · 27a03a3 · 27a03a3 · chriselrod · May 27, 2022
1 parent 27224a7
commit 27a03a3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "StrideArraysCore"
 uuid = "7792a7ef-975c-4747-a70f-980b88e8d1da"
 authors = ["Chris Elrod <[email protected]> and contributors"]
-version = "0.3.7"
+version = "0.3.8"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

diff --git a/src/thread_compatible.jl b/src/thread_compatible.jl
@@ -11,7 +11,13 @@ end
 @inline object_and_preserve(x::String) = (x, x)
 @inline object_and_preserve(A::AbstractArray{T}) where {T<:NativeTypes} =
   array_object_and_preserve(ArrayInterface.device(A), A)
-@inline object_and_preserve(A::AbstractArray) = (A, A)
+@inline function object_and_preserve(A::AbstractArray{T}) where {T}
+  if isbitstype(T) && ArrayInterface.device(A) === ArrayInterface.CPUPointer()
+    (PtrArray(A), preserve_buffer(A))
+  else
+    (A, A)
+  end
+end
 @inline function object_and_preserve(bc::Base.Broadcast.Broadcasted)
   if isbits(bc.f) && isbits(bc.axes)
     bca, bcp = object_and_preserve(bc.args)