From dcadae65f57b9000362f318b35846178c52ec1af Mon Sep 17 00:00:00 2001 From: Ioannis Magkanaris Date: Mon, 11 May 2026 16:24:28 +0200 Subject: [PATCH] Update thread block size to (256, 1) Co-authored-by: Daniel Ganellari --- .../runners/dace/transformations/auto_optimize.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/gt4py/next/program_processors/runners/dace/transformations/auto_optimize.py b/src/gt4py/next/program_processors/runners/dace/transformations/auto_optimize.py index 799e8ad228..62b658f839 100644 --- a/src/gt4py/next/program_processors/runners/dace/transformations/auto_optimize.py +++ b/src/gt4py/next/program_processors/runners/dace/transformations/auto_optimize.py @@ -113,8 +113,8 @@ def gt_auto_optimize( gpu: bool, unit_strides_kind: Optional[gtx_common.DimensionKind] = None, make_persistent: bool = False, - gpu_block_size: Optional[Sequence[int | str] | str] = (32, 8, 1), - gpu_block_size_1d: Optional[Sequence[int | str] | str] = (64, 1, 1), + gpu_block_size: Optional[Sequence[int | str] | str] = (256, 1, 1), + gpu_block_size_1d: Optional[Sequence[int | str] | str] = None, gpu_block_size_2d: Optional[Sequence[int | str] | str] = None, gpu_block_size_3d: Optional[Sequence[int | str] | str] = None, gpu_maxnreg: Optional[int] = None,