|
| 1 | +module AlignedArrays |
| 2 | + import Mmap |
| 3 | + |
| 4 | + |
| 5 | + export AlignedArray, AlignedVector, AlignedMatrix, AlignedVecOrMat |
| 6 | + export PageAlignedArray, PageAlignedVector, PageAlignedMatrix, PageAlignedVecOrMat |
| 7 | + |
| 8 | + |
| 9 | + const PAGESIZE = Mmap.PAGESIZE |
| 10 | + |
| 11 | + |
| 12 | + struct AlignedArray{T, N, A} <: DenseArray{T, N} |
| 13 | + parent::Array{T, N} |
| 14 | + addr::Ref{Ptr{Cvoid}} |
| 15 | + |
| 16 | + function AlignedArray{T, N, A}(::UndefInitializer, dims::NTuple{N, Integer}) where {T, N, A} |
| 17 | + ispow2(A) || error("Alignment must be a power of two") |
| 18 | + isconcretetype(T) || error("Element type must be a concrete type") |
| 19 | + |
| 20 | + size = isempty(dims) ? 0 : reduce(*, dims) |
| 21 | + @static if Sys.islinux() |
| 22 | + addr = Ref(C_NULL) |
| 23 | + ccall(:posix_memalign, Cint, (Ptr{Ptr{Cvoid}}, Csize_t, Csize_t), addr, A, size) == 0 || error("Failed to allocate aligned memory") |
| 24 | + else |
| 25 | + error("Operating system not yet supported") |
| 26 | + end |
| 27 | + |
| 28 | + a = new{T, N, A}(unsafe_wrap(Array{T, N}, reinterpret(Ptr{T}, addr[]), dims, own = false), addr) |
| 29 | + finalizer(a.addr) do x |
| 30 | + @static if Sys.islinux() |
| 31 | + ccall(:free, Cvoid, (Ptr{Cvoid},), x[]) |
| 32 | + end |
| 33 | + end |
| 34 | + return a |
| 35 | + end |
| 36 | + end |
| 37 | + |
| 38 | + AlignedArray{T, N, A}(u::UndefInitializer, dims::Integer...) where {T, N, A} = AlignedArray{T, N, A}(u, dims) |
| 39 | + |
| 40 | + const AlignedVector{T, A} = AlignedArray{T, 1, A} |
| 41 | + const AlignedMatrix{T, A} = AlignedArray{T, 2, A} |
| 42 | + const AlignedVecOrMat{T, A} = Union{AlignedVector{T, A}, AlignedMatrix{T, A}} |
| 43 | + |
| 44 | + const PageAlignedArray{T, N} = AlignedArray{T, N, PAGESIZE} |
| 45 | + const PageAlignedVector{T} = AlignedVector{T, PAGESIZE} |
| 46 | + const PageAlignedMatrix{T} = AlignedMatrix{T, PAGESIZE} |
| 47 | + const PageAlignedVecOrMat{T} = AlignedVecOrMat{T, PAGESIZE} |
| 48 | + |
| 49 | + |
| 50 | + Base.parent(a::AlignedArray) = a.parent |
| 51 | + |
| 52 | + Base.pointer(a::AlignedArray) = pointer(parent(a)) |
| 53 | + |
| 54 | + Base.size(a::AlignedArray) = size(parent(a)) |
| 55 | + Base.length(a::AlignedArray) = length(parent(a)) |
| 56 | + Base.axes(a::AlignedArray) = axes(parent(a)) |
| 57 | + |
| 58 | + Base.IndexStyle(::Type{A}) where {T, N, A<:AlignedArray{T, N}} = IndexStyle(Array{T, N}) |
| 59 | + Base.getindex(a::AlignedArray, args...) = getindex(parent(a), args...) |
| 60 | + Base.setindex!(a::AlignedArray, args...) = setindex!(parent(a), args...) |
| 61 | + Base.iterate(a::AlignedArray, args...) = iterate(parent(a), args...) |
| 62 | + |
| 63 | + Base.similar(a::AlignedArray, args...) = similar(parent(a), args...) |
| 64 | + |
| 65 | + Base.show(io::IO, m::MIME"text/plain", a::AlignedArray) = show(io, m, parent(a)) |
| 66 | +end |
0 commit comments