Skip to content

Commit 0af8688

Browse files
committed
test/stencil: Skip 4D stencil on GPUs
1 parent f595669 commit 0af8688

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

test/array/stencil.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import Dagger: @stencil, Wrap, Pad, Reflect, Clamp, LinearExtrapolate
22

3-
function test_stencil()
3+
function test_stencil(; gpu::Bool=false)
44
@testset "Simple assignment" begin
55
A = zeros(Blocks(2, 2), Int, 4, 4)
66
@stencil A[idx] = 1
@@ -387,6 +387,8 @@ function test_stencil()
387387

388388
# From issue #669
389389
for N in 3:4
390+
# Fine-grained 4D GPU stencils require too many halo copies for typical GPU memory
391+
gpu && N == 4 && continue
390392
@testset "$(N)D array" begin
391393
A = ones(Blocks(ntuple(_->1, N)...), Int, ntuple(_->3, N)...)
392394
Dagger.allowscalar() do
@@ -497,7 +499,7 @@ end
497499
kind == :oneAPI && continue
498500
@testset "$kind" begin
499501
Dagger.with_options(;scope) do
500-
test_stencil()
502+
test_stencil(; gpu=true)
501503
end
502504
end
503505
end

0 commit comments

Comments
 (0)