Skip to content

Commit e4b865b

Browse files
committed
[Metal] Use safe block_size
1 parent 5f35417 commit e4b865b

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)