Skip to content

Commit

Permalink
#8361: add reshard sweep test
Browse files Browse the repository at this point in the history
  • Loading branch information
ntarafdar committed May 10, 2024
1 parent d0022a6 commit 1d7edb4
Showing 1 changed file with 1 addition and 27 deletions.
28 changes: 1 addition & 27 deletions tests/ttnn/sweep_tests/sweeps/reshard_height_width.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,33 +45,7 @@
}


def skip(
*,
layout,
height,
width,
input_num_cores_x,
input_num_cores_y,
input_shard_strategy,
output_num_cores_x,
output_num_cores_y,
output_shard_strategy,
**_,
) -> Tuple[bool, Optional[str]]:
input_num_cores = input_num_cores_x * input_num_cores_y
if layout == ttnn.TILE_LAYOUT:
if height % (input_num_cores) != 0:
return True, "Input Shard not divisible"
if width % (input_num_cores) != 0:
return True, "Input Shard not divisible"

output_num_cores = output_num_cores_x * output_num_cores_y
if layout == ttnn.TILE_LAYOUT:
if height % (output_num_cores) != 0:
return True, "Output Shard not divisible"
if width % (output_num_cores) != 0:
return True, "Output Shard not divisible"

def skip(**_) -> Tuple[bool, Optional[str]]:
return False, None


Expand Down

0 comments on commit 1d7edb4

Please sign in to comment.