|
21 | 21 |
|
22 | 22 | ## Memory Operations |
23 | 23 |
|
24 | | -KA.allocate(::POCLBackend, ::Type{T}, dims::Tuple) where {T} = Array{T}(undef, dims) |
| 24 | +KA.allocate(::POCLBackend, ::Type{T}, dims::Tuple; unified::Bool = false) where {T} = Array{T}(undef, dims) |
25 | 25 |
|
26 | | -function KA.zeros(backend::POCLBackend, ::Type{T}, dims::Tuple) where {T} |
27 | | - arr = KA.allocate(backend, T, dims) |
| 26 | +function KA.zeros(backend::POCLBackend, ::Type{T}, dims::Tuple; kwargs...) where {T} |
| 27 | + arr = KA.allocate(backend, T, dims; kwargs...) |
28 | 28 | kernel = KA.init_kernel(backend) |
29 | 29 | kernel(arr, zero, T, ndrange = length(arr)) |
30 | 30 | return arr |
31 | 31 | end |
32 | | -function KA.ones(backend::POCLBackend, ::Type{T}, dims::Tuple) where {T} |
33 | | - arr = KA.allocate(backend, T, dims) |
| 32 | +function KA.ones(backend::POCLBackend, ::Type{T}, dims::Tuple; kwargs...) where {T} |
| 33 | + arr = KA.allocate(backend, T, dims; kwargs...) |
34 | 34 | kernel = KA.init_kernel(backend) |
35 | 35 | kernel(arr, one, T; ndrange = length(arr)) |
36 | 36 | return arr |
@@ -58,6 +58,7 @@ KA.pagelock!(::POCLBackend, x) = nothing |
58 | 58 | KA.get_backend(::Array) = POCLBackend() |
59 | 59 | KA.synchronize(::POCLBackend) = nothing |
60 | 60 | KA.supports_float64(::POCLBackend) = true |
| 61 | +KA.supports_unified(::POCLBackend) = true |
61 | 62 |
|
62 | 63 |
|
63 | 64 | ## Kernel Launch |
|
0 commit comments