Skip to content

Commit 5822449

Browse files
committed
Add note
1 parent 432d2d4 commit 5822449

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch_ttnn/passes/lowering/to_tt_pass.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -508,6 +508,7 @@ def lower_binary_eltwise(fn, args):
508508
grid_size_x = device.compute_with_storage_grid_size().x
509509
grid_size_y = device.compute_with_storage_grid_size().y
510510
shard_shape = N * H * W // grid_size_x, C // grid_size_y
511+
# TODO: Add support for shard_shape = 0
511512
if shard_shape[0] == 0 or shard_shape[1] == 0:
512513
return None
513514
# input tensor

0 commit comments

Comments
 (0)