Skip to content

Commit 2b39025

Browse files
committed
Actually pass neutral element into scan
1 parent 0c00b83 commit 2b39025

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

src/accumulate.jl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -200,17 +200,17 @@ end
200200

201201
## Base interface
202202

203-
Base._accumulate!(op, output::AnyCuArray, input::AnyCuVector, dims::Nothing, init::Nothing) =
204-
scan!(op, output, input; dims=1)
203+
Base._accumulate!(op, output::AnyCuArray, input::AnyCuVector, dims::Nothing, init::Nothing, kwargs...) =
204+
scan!(op, output, input; dims=1, kwargs...)
205205

206-
Base._accumulate!(op, output::AnyCuArray, input::AnyCuArray, dims::Integer, init::Nothing) =
207-
scan!(op, output, input; dims=dims)
206+
Base._accumulate!(op, output::AnyCuArray, input::AnyCuArray, dims::Integer, init::Nothing, kwargs...) =
207+
scan!(op, output, input; dims=dims, kwargs...)
208208

209-
Base._accumulate!(op, output::AnyCuArray, input::CuVector, dims::Nothing, init::Some) =
210-
scan!(op, output, input; dims=1, init=init)
209+
Base._accumulate!(op, output::AnyCuArray, input::CuVector, dims::Nothing, init::Some, kwargs...) =
210+
scan!(op, output, input; dims=1, init=init, kwargs...)
211211

212-
Base._accumulate!(op, output::AnyCuArray, input::AnyCuArray, dims::Integer, init::Some) =
213-
scan!(op, output, input; dims=dims, init=init)
212+
Base._accumulate!(op, output::AnyCuArray, input::AnyCuArray, dims::Integer, init::Some, kwargs...) =
213+
scan!(op, output, input; dims=dims, init=init, kwargs...)
214214

215215
Base.accumulate_pairwise!(op, result::AnyCuVector, v::AnyCuVector) = accumulate!(op, result, v)
216216

0 commit comments

Comments
 (0)