Skip to content

Commit

Permalink
#13142: Add documentation for memory config
Browse files Browse the repository at this point in the history
  • Loading branch information
mcw-anasuya committed Sep 26, 2024
1 parent a181050 commit 6c75d6c
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions ttnn/ttnn/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,6 +98,9 @@ def create_sharded_memory_config(
halo (bool, optional): if the shards have overlapping values. Defaults to `False`.
use_height_and_width_as_shard_shape (bool, optional): if True, the height and width of the tensor will be used as the shard shape. Defaults to `False`. If is False, the shard shape will be calculated based on the core_grid and the tensor shape where tensor shape is seen as [math.prod(dims), width]
Returns:
ttnn.MemoryConfig: the MemoryConfig object.
Example:
>>> tensor = ttnn.create_sharded_memory_config((5, 8), (320,64), ttnn.ShardStrategy.BLOCK, ttnn.ShardOrientation.ROW_MAJOR, False)
Expand Down

0 comments on commit 6c75d6c

Please sign in to comment.