Optimize conversion with no size change

JuliaArrays · Jul 1, 2022 · 53e8df5 · 53e8df5
1 parent 3778fa9
commit 53e8df5
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/unsafe_array.jl b/src/unsafe_array.jl
@@ -149,12 +149,16 @@ end
 Base.deepcopy(A::UnsafeArray) = copyto!(similar(A), A)
 
 function Base.reinterpret(::Type{DST}, A::UnsafeArray{SRC}) where {DST, SRC}
-    sz = size(A)
-    sz1, rem = divrem(sz[1] * sizeof(SRC), sizeof(DST))
-    @boundscheck if rem != zero(rem)
-        throw(ArgumentError("Resulting array would have non-integral first dimension"))
+    if sizeof(DST) != sizeof(SRC)
+        sz = size(A)
+        sz1, rem = divrem(sz[1] * sizeof(SRC), sizeof(DST))
+        @boundscheck if rem != zero(rem)
+            throw(ArgumentError("Resulting array would have non-integral first dimension"))
+        end
+        UnsafeArray(convert(Ptr{DST}, pointer(A)), (sz1, Base.tail(sz)...))
+    else
+        UnsafeArray(convert(Ptr{DST}, pointer(A)), size(A))
     end
-    UnsafeArray(convert(Ptr{DST}, pointer(A)), (sz1, Base.tail(sz)...))
 end
 
 # # Defining Base.unaliascopy results in very bad broadcast performance for