Skip to content

Commit f41d1e0

Browse files
committed
More fix
1 parent cc240c6 commit f41d1e0

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/pocl/backend.jl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ KA.functional(::POCLBackend) = true
5656
KA.pagelock!(::POCLBackend, x) = nothing
5757

5858
KA.get_backend(::Array) = POCLBackend()
59-
KA.synchronize(::POCLBackend) = nothing
59+
KA.synchronize(::POCLBackend) = cl.finish(cl.queue())
6060
KA.supports_float64(::POCLBackend) = true
6161
KA.supports_unified(::POCLBackend) = true
6262

@@ -160,13 +160,14 @@ end
160160

161161

162162
function KI.kernel_max_work_group_size(::POCLBackend, kikern::KI.KIKernel{<:POCLBackend}; max_work_items::Int=typemax(Int))::Int
163-
4096
163+
wginfo = cl.work_group_info(kikern.kern.fun, device())
164+
Int(min(wginfo.size, max_work_items))
164165
end
165166
function KI.max_work_group_size(::POCLBackend)::Int
166-
4096
167+
Int(device().max_work_group_size)
167168
end
168169
function KI.multiprocessor_count(::POCLBackend)::Int
169-
1
170+
Int(device().max_compute_units)
170171
end
171172

172173
## Indexing Functions

test/intrinsics.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,7 @@ function intrinsics_testsuite(backend, AT)
3030
# Test with small kernel
3131
N = 16
3232
results = AT(zeros(Int, 6, N))
33-
34-
kernel = KI.@kikernel backend() test_intrinsics_kernel(results)
33+
kernel = KI.@kikernel backend() launch=false test_intrinsics_kernel(results)
3534

3635
@test KI.kernel_max_work_group_size(backend(), kernel) isa Int
3736
@test KI.kernel_max_work_group_size(backend(), kernel; max_work_items=1) == 1

0 commit comments

Comments
 (0)