tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.

https://docs.tenstorrent.com/ttnn/latest/index.html

Apache License 2.0

485 stars 79 forks source link

#14974: ttnn::{full,empty}_like Tensor creation API for MeshDevice #15333

Open omilyutin-tt opened 4 days ago

omilyutin-tt commented 4 days ago

Ticket

14974

Problem description

Extensions to Tensor creation APIs to support MeshDevice.

What's changed

Follow up from https://github.com/tenstorrent/tt-metal/pull/15191:

Extend support for MeshDevice in ttnn::empty_like, ttnn::full, ttnn::full_like.
Minor refactor of Tensor allocation functions.

Checklist

[ ] Post commit CI passes
[ ] Blackhole Post commit (if applicable)
[ ] Model regression CI testing passes (if applicable)
[ ] Device performance regression CI testing passes (if applicable)
[ ] New/Existing tests provide coverage for changes

omilyutin-tt commented 3 hours ago

Thanks @rfurko-tt!