tenstorrent / tt-metal

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

softshrink_bw operation does not work bfloat8_b #13927

Open nemanjagrujic opened 1 week ago

nemanjagrujic commented 1 week ago

softshrink_bw operation does not work when input tensor is bfloat8_b. Documentation does not mention that limitation.

Steps to reproduce

You can reproduce this error with sweep test tests/sweep_framework/sweeps/eltwise/unary_backward/softshrink_bw.py. To run:

python3 tests/sweep_framework/runner.py --elastic cloud --module-name eltwise.unary_backward.softshrink_bw

Results can be found on elastic cloud as explained here: https://github.com/tenstorrent/tt-metal/tree/main/tests/sweep_framework

KalaivaniMCW commented 3 days ago

Documentation has been updated

KalaivaniMCW commented 3 days ago
image