search

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.
Apache License 2.0
471 stars 75 forks source link

Sweep TM Ops #11621

Open sjameelTT opened 2 months ago

sjameelTT commented 2 months ago

We need to add sweeps for each of the ttnn ops to determine what's supported and what's not supported. This is the first step on the road to achieving generality. TM Ops are all the ones in the following folder ttnn/cpp/ttnn/operations/data_movement

PyTorch Ops in Traces (High Priority) Op Name Status Pass Rate (%) Number of Tests Issue Number Owner
Copy In Progress #12014 @jaykru-tt
Slice In Progress 74.8(T) 53.54(R) #12035 @sjameelTT
Fill RM Not Started #12019 @sjameelTT
Fill Ones RM Not Started #12020 @sjameelTT
Pad Not Started #12025 @sjameelTT
Permute Not Started 27.99(T) 46.55(R) #12026 @sjameelTT
Repeat Not Started #12027 @tarafdarTT
Repeat Interleave Not Started #12028 @tarafdarTT
Reshape on Device Not Started #12029 @tarafdarTT
Reshape Not Started #12030 @tarafdarTT
Squeeze Not Started #13181 @tt-atulk
Unsqueeze Not Started #13182 @tt-atulk
Infrastructure Ops (High Priority) Op Name Status Pass Rate (%) Number of Tests Issue Number Owner
Reshard Not Started #12480 @tarafdarTT
Interleaved to Sharded Not Started #12031 @tarafdarTT
Sharded to Interleaved Not Started #12032 @tarafdarTT
Interleaved to Sharded Partial Not Started #12033 @tarafdarTT
Sharded to Interleaved Partial Not Started #12034 @tarafdarTT
Tilize Not Started #12037 @jaykru-tt
Tilize with val padding Not Started #12038 @tarafdarTT
Tilize with Zero Padding Not Started #12039 @tarafdarTT
Transpose Not Started #12040 @sjameelTT
Untilize Not Started #12041 @jaykru-tt
Untilize with Halo Not Started #12042 @jaykru-tt
Untilize with Unpadding Not Started #12043 @jaykru-tt

Other PyTorch Ops (Low priority)

Op Name Status Pass Rate (%) Number of Tests Issue Number Owner
Concat Complete 50.19 28106 #11622 @sjameelTT
Clone Not Started #12015 @jaykru-tt
Assign Not Started #12016 @jaykru-tt
Data Transfer to Host Not Started #12017 @jaykru-tt
Data Transfer to Device Not Started #12018 @jaykru-tt
Fold Not Started #12021 @tarafdarTT
Indexed Fill Not Started #12022 @sjameelTT
Move Not Started #12023 @sjameelTT
Nonzero Not Started #12024 @sjameelTT
Split Not Started #12036 @jaykru-tt
sjameelTT commented 1 month ago
sjameelTT commented 1 month ago

https://github.com/tenstorrent/tt-metal/issues/13181 https://github.com/tenstorrent/tt-metal/issues/13182