Skip to content

Commit 111c89b

Browse files
authored
Merge pull request #36 from christiangnrd/metal
[Metal] Use safe `block_size` in `accumulate`
2 parents 5f35417 + e4b865b commit 111c89b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ext/AcceleratedKernelsMetalExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ function AK.accumulate!(
1818
alg::AK.AccumulateAlgorithm=AK.ScanPrefixes(),
1919

2020
# GPU settings
21-
block_size::Int=1024,
21+
block_size::Int=256,
2222
temp::Union{Nothing, AbstractArray}=nothing,
2323
temp_flags::Union{Nothing, AbstractArray}=nothing,
2424
)
@@ -43,7 +43,7 @@ function AK.accumulate!(
4343
alg::AK.AccumulateAlgorithm=AK.ScanPrefixes(),
4444

4545
# GPU settings
46-
block_size::Int=1024,
46+
block_size::Int=256,
4747
temp::Union{Nothing, AbstractArray}=nothing,
4848
temp_flags::Union{Nothing, AbstractArray}=nothing,
4949
)

0 commit comments

Comments
 (0)