Skip to content

Commit

Permalink
Use JLArray to test
Browse files Browse the repository at this point in the history
  • Loading branch information
vchuravy authored and maleadt committed Oct 30, 2024
1 parent d7f487b commit 2314e24
Showing 1 changed file with 0 additions and 9 deletions.
9 changes: 0 additions & 9 deletions lib/JLArrays/src/JLArrays.jl
Original file line number Diff line number Diff line change
Expand Up @@ -332,15 +332,6 @@ end
Adapt.adapt_storage(::Adaptor, x::JLArray{T,N}) where {T,N} =
JLDeviceArray{T,N}(x.data[], x.offset, x.dims)

function GPUArrays.mapreducedim!(f, op, R::AnyJLArray, A::Union{AbstractArray,Broadcast.Broadcasted};
init=nothing)
if init !== nothing
fill!(R, init)
end
@allowscalar Base.reducedim!(op, typed_data(R), map(f, A))
R
end

## KernelAbstractions interface

KernelAbstractions.get_backend(a::JLA) where JLA <: JLArray = JLBackend()
Expand Down

0 comments on commit 2314e24

Please sign in to comment.