|  | 
| 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