From a71e0d318a6f96bef38ce4b7b260d8f3fd73d91d Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Mon, 10 Feb 2025 18:03:52 +0100
Subject: [PATCH] Temporarily change default CUDA block size for CUDA jit
---
src/pystencils/jit/gpu_cupy.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pystencils/jit/gpu_cupy.py b/src/pystencils/jit/gpu_cupy.py
index 54fb41173..331b58ce5 100644
--- a/src/pystencils/jit/gpu_cupy.py
+++ b/src/pystencils/jit/gpu_cupy.py
@@ -242,7 +242,7 @@ class CupyKernelWrapper(KernelWrapper):
class CupyJit(JitBase):
- def __init__(self, default_block_size: Sequence[int] = (128, 2, 1)):
+ def __init__(self, default_block_size: Sequence[int] = (128, 1, 1)):
self._runtime_headers = {"<cstdint>"}
if len(default_block_size) > 3:
--
GitLab