TTNN demo for UNet - Githubissues

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

497 stars 83 forks source link

TTNN demo for UNet #5792

Open dvartaniansTT opened 9 months ago

dvartaniansTT commented 9 months ago

Is your feature request related to a problem? Please describe. have a demo for UNet running on device via TTNN

Describe the solution you'd like Have a demo of UNet carvana and/or UNet Brain-segmentation running on device via TTNN. carvana and Brain MRI segmentation via UNet

dvartaniansTT commented 8 months ago

currently blocked by: 5857 & 6326

dvartaniansTT commented 8 months ago

a quick summary for the brain-mri variant here:

we have unit tests for all required convs: conv2d brain unit test issue 5857
we have unit test for all required maxpool_2d: maxpool2d brain UNet unit test (all pass :rocket:)
I can't write unit tests for the concat ops yet as I'm blocked by convTraspose2D (the op before concats) to know the sharding shapes needed for the concats.
I have filed an issue for convTranspose2d bringup: #6326
I also added a unit test for log sigmoid in TTNN: log sigmoid TTNN unit test (pass :rocket:)

dvartaniansTT commented 8 months ago

currently blocked by: https://github.com/tenstorrent-metal/tt-metal/issues/5857 & https://github.com/tenstorrent-metal/tt-metal/issues/6326 once these 2 issues are resolved, I can test concats and then the model end to end.