Skip to content

Commit

Permalink
#13127: Add support for sharding + alignment in TensorLayout and tens…
Browse files Browse the repository at this point in the history 
…or creation

- Add new TensorLayout constructor that takes in optional shard shape and 2D shard alignment
- Add tests to test basic tensor creation with sharding + alignment
  • Loading branch information
@TT-BrianLiu
TT-BrianLiu committed Nov 6, 2024
1 parent 7b6c0a6 commit 4d32ca5
Show file tree
Hide file tree
Showing 4 changed files with 406 additions and 52 deletions.
Loading

0 comments on commit 4d32ca5

Please sign in to comment.