Skip to content

Commit

Permalink
one more ShardSpec update
Browse files Browse the repository at this point in the history
  • Loading branch information
kmabeeTT committed Jan 14, 2025
1 parent 5dfa5f3 commit a7ac684
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lib/OpModel/TTNN/Conversion.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ getShardSpec(const mlir::tt::ttnn::TTNNLayoutAttr &layout) {
return isShardedMemoryLayout(layout.getMemLayout().getValue())
? std::make_optional(ShardSpec(getCoreRangeSet(layout),
getShardShape(layout),
ShardOrientation::ROW_MAJOR, false))
ShardOrientation::ROW_MAJOR))
: std::nullopt;
}

Expand Down

0 comments on commit a7ac684

Please sign in to comment.