@neighbors fails for 3D cube
When extending the official 2D @stencil + @neighbors example to 3D (i.e. a cubic DArray), the code errors.
```
using Dagger
import Dagger: @stencil, Wrap
A = ones(Blocks(1, 1, 1), Int, 3, 3, 3)
A[:] = 1:length(A)
B = zeros(Blocks(1, 1, 1), Int, 3, 3, 3)
Dagger.spawn_datadeps() do
@stencil begin
B[idx] = sum(@neighbors(A[idx], 1, Wrap())) / 27
end
end
ERROR: DTaskFailedException:
Root Exception Type: AssertionError
Root Exception:
AssertionError: Halo mismatch: edges=6 corners=6
```
关闭于 2026-02-01 8 条评论