Skip to content
Snippets Groups Projects
Commit 466e3602 authored by Frederik Hennig's avatar Frederik Hennig
Browse files

fix 2d indexing test

parent a52c20a4
No related branches found
No related tags found
1 merge request!457Fix Linear3D Indexing if Iteration Space Rank < 3
Pipeline #75683 passed
...@@ -99,17 +99,17 @@ def test_indexing_options_2d( ...@@ -99,17 +99,17 @@ def test_indexing_options_2d(
ast = create_kernel(asm, cfg) ast = create_kernel(asm, cfg)
kernel = ast.compile() kernel = ast.compile()
src_arr = cp.ones((18, 34)) src_arr = cp.ones((18, 42))
dst_arr = cp.zeros_like(src_arr) dst_arr = cp.zeros_like(src_arr)
if manual_grid: if manual_grid:
match indexing_scheme: match indexing_scheme:
case "linear3d": case "linear3d":
kernel.launch_config.block_size = (10, 8, 1) 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": case "blockwise4d":
kernel.launch_config.block_size = (40, 1, 1) 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": elif indexing_scheme == "linear3d":
kernel.launch_config.block_size = (10, 8, 1) kernel.launch_config.block_size = (10, 8, 1)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment