Skip to content

Commit

Permalink
Modify cases
Browse files Browse the repository at this point in the history
  • Loading branch information
Aswinmcw committed Jan 30, 2025
1 parent 397737d commit cc79106
Showing 1 changed file with 0 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -157,24 +157,18 @@ def test_all_gather_async_tg(
"per_chip_output_shape, dim, layout",
[
([1, 1, 32, 32], 3, ttnn.TILE_LAYOUT),
# ([1, 1, 32, 32 * 2], 3, ttnn.TILE_LAYOUT),
# ([1, 1, 64, 32], 3, ttnn.TILE_LAYOUT),
# ([1, 1, 64, 64], 3, ttnn.TILE_LAYOUT),
# ([1, 1, 128, 128], 0, ttnn.TILE_LAYOUT),
],
)
@pytest.mark.parametrize(
"input_dtype",
[
ttnn.bfloat16,
# ttnn.bfloat8_b,
],
)
@pytest.mark.parametrize(
"mem_config",
[
ttnn.MemoryConfig(buffer_type=ttnn.BufferType.DRAM),
# ttnn.MemoryConfig(buffer_type=ttnn.BufferType.L1),
],
)
@pytest.mark.parametrize("math_op", [ttnn.ReduceType.Sum])
Expand Down

0 comments on commit cc79106

Please sign in to comment.