Skip to content

Commit 8552cc5

Browse files
committed
fixup! datadeps: Add at-stencil helper
1 parent 963f952 commit 8552cc5

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/stencil.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,12 @@ function load_neighbor_edge(arr, dim, dir, neigh_dist)
1212
stop_idx = CartesianIndex(ntuple(i -> i == dim ? (firstindex(arr, i) + neigh_dist - 1) : lastindex(arr, i), ndims(arr)))
1313
end
1414
# FIXME: Don't collect
15-
return move(thunk_processor(), collect(@view arr[start_idx:stop_idx]))
15+
return move(task_processor(), collect(@view arr[start_idx:stop_idx]))
1616
end
1717
function load_neighbor_corner(arr, corner_side, neigh_dist)
1818
start_idx = CartesianIndex(ntuple(i -> corner_side[i] == 0 ? (lastindex(arr, i) - neigh_dist + 1) : firstindex(arr, i), ndims(arr)))
1919
stop_idx = CartesianIndex(ntuple(i -> corner_side[i] == 0 ? lastindex(arr, i) : (firstindex(arr, i) + neigh_dist - 1), ndims(arr)))
20-
return move(thunk_processor(), collect(@view arr[start_idx:stop_idx]))
20+
return move(task_processor(), collect(@view arr[start_idx:stop_idx]))
2121
end
2222
function select_neighborhood_chunks(chunks, idx, neigh_dist, boundary)
2323
@assert neigh_dist isa Integer && neigh_dist > 0 "Neighborhood distance must be an Integer greater than 0"
@@ -87,7 +87,7 @@ function load_neighborhood(arr::HaloArray{T,N}, idx) where {T,N}
8787
return @view arr[start_idx:stop_idx]
8888
end
8989
function inner_stencil!(f, output, read_vars)
90-
processor = thunk_processor()
90+
processor = task_processor()
9191
inner_stencil_proc!(processor, f, output, read_vars)
9292
end
9393
# Non-KA (for CPUs)
@@ -121,14 +121,14 @@ function load_boundary_edge(pad::Pad, arr, dim, dir, neigh_dist)
121121
end
122122
edge_size = ntuple(i -> length(start_idx[i]:stop_idx[i]), ndims(arr))
123123
# FIXME: return Fill(pad.padval, edge_size)
124-
return move(thunk_processor(), fill(pad.padval, edge_size))
124+
return move(task_processor(), fill(pad.padval, edge_size))
125125
end
126126
function load_boundary_corner(pad::Pad, arr, corner_side, neigh_dist)
127127
start_idx = CartesianIndex(ntuple(i -> corner_side[i] == 0 ? (lastindex(arr, i) - neigh_dist + 1) : firstindex(arr, i), ndims(arr)))
128128
stop_idx = CartesianIndex(ntuple(i -> corner_side[i] == 0 ? lastindex(arr, i) : (firstindex(arr, i) + neigh_dist - 1), ndims(arr)))
129129
corner_size = ntuple(i -> length(start_idx[i]:stop_idx[i]), ndims(arr))
130130
# FIXME: return Fill(pad.padval, corner_size)
131-
return move(thunk_processor(), fill(pad.padval, corner_size))
131+
return move(task_processor(), fill(pad.padval, corner_size))
132132
end
133133

134134
"""

0 commit comments

Comments
 (0)