Skip to content
Snippets Groups Projects
Commit 1279fb9a authored by Wassim KABALAN's avatar Wassim KABALAN
Browse files

Increase test_transpose size to work on 8 devices

parent 7df2e642
No related branches found
No related tags found
No related merge requests found
......@@ -25,7 +25,7 @@ pencil_1 = (size // 2, size // (size // 2)) # 2x2 for V100 and 4x2 for A100
pencil_2 = (size // (size // 2), size // 2) # 2x2 for V100 and 2x4 for A100
decomp = [(size, 1), (1, size), pencil_1, pencil_2]
global_shapes = [(4, 8, 16), (4, 4, 4), (29 * size, 19 * size, 17 * size)]
global_shapes = [(8, 16, 32), (8, 8, 8), (29 * size, 19 * size, 17 * size)]
# Cubes, non-cubes and primes
local_transpose = [False, True]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment