Skip to content

Commit

Permalink
Remove commented shapes
Browse files Browse the repository at this point in the history
  • Loading branch information
Aswinmcw committed Jan 30, 2025
1 parent 222879c commit d51707d
Showing 1 changed file with 0 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,22 +18,16 @@
"num_devices, num_links, output_shape, dim, layout",
[
(4, 1, [1, 1, 64, 512], 3, ttnn.TILE_LAYOUT),
# (4, 1, [1, 1, 32, 32768], 3, ttnn.TILE_LAYOUT),
# (4, 1, [1, 1, 1024, 1024], 3, ttnn.TILE_LAYOUT),
# (4, 1, [1, 1, 2048, 16384], 3, ttnn.TILE_LAYOUT),
(4, 1, [1, 1, 32, 1280], 0, ttnn.TILE_LAYOUT),
(4, 1, [1, 1, 32, 7168], 0, ttnn.TILE_LAYOUT),
(8, 1, [1, 1, 32, 2048], 0, ttnn.TILE_LAYOUT),
(4, 1, [1, 1, 32, 3584], 0, ttnn.TILE_LAYOUT),
(4, 1, [1, 1, 32, 32], 0, ttnn.TILE_LAYOUT),
# (4, 1, [1, 1, 8, 32], 2, ttnn.TILE_LAYOUT),
],
)
@pytest.mark.parametrize(
"input_dtype",
[
ttnn.bfloat16,
# ttnn.bfloat8_b,
],
)
@pytest.mark.parametrize(
Expand Down

0 comments on commit d51707d

Please sign in to comment.