Skip to content

Commit a647992

Browse files
committed
Limit groupsize to 256 for oneAPI
1 parent 7c923fb commit a647992

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

test/groupreduce.jl

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,12 @@ end
1313
end
1414

1515
function groupreduce_testsuite(backend, AT)
16+
# TODO should be better way of querying max groupsize
17+
groupsizes = "$backend" == "oneAPIBackend" ?
18+
(256,) :
19+
(256, 512, 1024)
1620
@testset "@groupreduce" begin
17-
@testset "thread reduction T=$T, n=$n" for T in (Float16, Float32, Int32, Int64), n in (256, 512, 1024)
21+
@testset "thread reduction T=$T, n=$n" for T in (Float16, Float32, Int32, Int64), n in groupsizes
1822
x = AT(ones(T, n))
1923
y = AT(zeros(T, 1))
2024

@@ -30,8 +34,7 @@ function groupreduce_testsuite(backend, AT)
3034

3135
warp_reduction = KernelAbstractions.supports_warp_reduction(backend())
3236
if warp_reduction
33-
@testset "warp reduction T=$T, n=$n" for T in (Float16, Float32, Int32, Int64), n in (256, 512, 1024)
34-
37+
@testset "warp reduction T=$T, n=$n" for T in (Float16, Float32, Int32, Int64), n in groupsizes
3538
x = AT(ones(T, n))
3639
y = AT(zeros(T, 1))
3740
groupreduce_1!(backend(), n)(y, x, +, zero(T), Reduction.warp; ndrange = n)

0 commit comments

Comments
 (0)