From 466e36021c72926f5fa2659b2cb2eedbd53068e9 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Tue, 11 Mar 2025 11:20:42 +0100
Subject: [PATCH] fix 2d indexing test
---
tests/kernelcreation/test_gpu.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/tests/kernelcreation/test_gpu.py b/tests/kernelcreation/test_gpu.py
index 45acaf5f8..f1905b1fc 100644
--- a/tests/kernelcreation/test_gpu.py
+++ b/tests/kernelcreation/test_gpu.py
@@ -99,17 +99,17 @@ def test_indexing_options_2d(
ast = create_kernel(asm, cfg)
kernel = ast.compile()
- src_arr = cp.ones((18, 34))
+ src_arr = cp.ones((18, 42))
dst_arr = cp.zeros_like(src_arr)
if manual_grid:
match indexing_scheme:
case "linear3d":
kernel.launch_config.block_size = (10, 8, 1)
- kernel.launch_config.grid_size = (4, 4, 1)
+ kernel.launch_config.grid_size = (4, 2, 1)
case "blockwise4d":
kernel.launch_config.block_size = (40, 1, 1)
- kernel.launch_config.grid_size = (32, 1, 1)
+ kernel.launch_config.grid_size = (16, 1, 1)
elif indexing_scheme == "linear3d":
kernel.launch_config.block_size = (10, 8, 1)
--
GitLab