tenstorrent / tt-mlir

Tenstorrent MLIR compiler
https://tenstorrent.github.io/tt-mlir/
Apache License 2.0
72 stars 9 forks source link

Implement mapping from GridAttr to TTNN grid #594

Open odjuricicTT opened 2 months ago

odjuricicTT commented 2 months ago

In TTIR we use virtual 1x64 grids to represent width / height sharding. This needs to be mapped to actual cores in TTNN.

Example: 1x11 grid does not map to a rect.