search

tenstorrent / tt-metal

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

ttnn.softmax returns zero valued tensor in convnet mnist model #12664

Open vigneshkeerthivasanx opened 2 weeks ago

vigneshkeerthivasanx commented 2 weeks ago

Describe the bug ttnn.softmax returns zero valued tensor in convnet mnist model resulting in PCC = 0.0.

To Reproduce Pick the test from commit Run pytest

Expected behavior As the above shared configuration works in pytorch, it is expected to run in ttnn as well.

Screenshots

 ERROR    | models.utility_functions:comp_pcc:602 - One tensor is all zero
E       AssertionError: 0.0
vigneshkeerthivasanx commented 1 week ago

CC: @boris-drazic @jvasilje