Skip to content

Commit 92141a7

Browse files
awadell1maleadt
authored andcommitted
Swap total_threads with elements
gpu_call signiture changed by #367 to rename "total_threads" to "elements"
1 parent 2d5776a commit 92141a7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: src/host/base.jl

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ function repeat_inner(xs::AnyGPUArray, inner)
2323
out = similar(xs, eltype(xs), inner .* size(xs))
2424
any(==(0), size(out)) && return out # consistent with `Base.repeat`
2525

26-
gpu_call(repeat_inner_kernel!, xs, inner, out; total_threads=prod(size(out)))
26+
gpu_call(repeat_inner_kernel!, xs, inner, out; elements=prod(size(out)))
2727
return out
2828
end
2929

@@ -47,7 +47,7 @@ function repeat_outer(xs::AnyGPUArray, outer)
4747
out = similar(xs, eltype(xs), outer .* size(xs))
4848
any(==(0), size(out)) && return out # consistent with `Base.repeat`
4949

50-
gpu_call(repeat_outer_kernel!, xs, size(xs), outer, out; total_threads=prod(size(out)))
50+
gpu_call(repeat_outer_kernel!, xs, size(xs), outer, out; elements=prod(size(out)))
5151
return out
5252
end
5353

0 commit comments

Comments
 (0)