File tree Expand file tree Collapse file tree 2 files changed +6
-6
lines changed
Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -56,7 +56,7 @@ KA.functional(::POCLBackend) = true
5656KA. pagelock! (:: POCLBackend , x) = nothing
5757
5858KA. get_backend (:: Array ) = POCLBackend ()
59- KA. synchronize (:: POCLBackend ) = nothing
59+ KA. synchronize (:: POCLBackend ) = cl . finish (cl . queue ())
6060KA. supports_float64 (:: POCLBackend ) = true
6161KA. supports_unified (:: POCLBackend ) = true
6262
@@ -160,13 +160,14 @@ end
160160
161161
162162function 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))
164165end
165166function KI. max_work_group_size (:: POCLBackend ):: Int
166- 4096
167+ Int ( device () . max_work_group_size)
167168end
168169function KI. multiprocessor_count (:: POCLBackend ):: Int
169- 1
170+ Int ( device () . max_compute_units)
170171end
171172
172173# # Indexing Functions
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments