From e4b865bbf089c337c522b59499f07fcc8b2f31b7 Mon Sep 17 00:00:00 2001
From: Christian Guinard <28689358+christiangnrd@users.noreply.github.com>
Date: Tue, 20 May 2025 10:43:23 -0300
Subject: [PATCH] [Metal] Use safe `block_size`

---
 ext/AcceleratedKernelsMetalExt.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ext/AcceleratedKernelsMetalExt.jl b/ext/AcceleratedKernelsMetalExt.jl
index 0f66cf3..245806b 100644
--- a/ext/AcceleratedKernelsMetalExt.jl
+++ b/ext/AcceleratedKernelsMetalExt.jl
@@ -18,7 +18,7 @@ function AK.accumulate!(
     alg::AK.AccumulateAlgorithm=AK.ScanPrefixes(),
 
     # GPU settings
-    block_size::Int=1024,
+    block_size::Int=256,
     temp::Union{Nothing, AbstractArray}=nothing,
     temp_flags::Union{Nothing, AbstractArray}=nothing,
 )
@@ -43,7 +43,7 @@ function AK.accumulate!(
     alg::AK.AccumulateAlgorithm=AK.ScanPrefixes(),
 
     # GPU settings
-    block_size::Int=1024,
+    block_size::Int=256,
     temp::Union{Nothing, AbstractArray}=nothing,
     temp_flags::Union{Nothing, AbstractArray}=nothing,
 )