tenstorrent / tt-mlir

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

Bringup Arange #1332

Closed LPanosTT closed 4 hours ago

LPanosTT commented 3 days ago

Issues: https://github.com/tenstorrent/tt-mlir/issues/1011, https://github.com/tenstorrent/tt-mlir/issues/1010 Add conversion patterns from stablehlo.iota and stablehlo.dynamic_iota to ttir.arange

Add pattern in TTIRToTTIRDecompositionPass to rewrite all ttir.arange ops where the arange_dimension is not the right-most dim. This has the effect of making-explicit the broadcasts and tms that would need to be done after executin ttnn.arange

add runtime support and basic silicon test