Open saichandax opened 5 months ago
Added unit tests for all the supported ttnn ops and tested them.
The result of the unit tests for the ops in mobilenetv2 are as follows:
a) TT_THROW @ tt_metal/impl/allocator/allocator.cpp:107: tt::exception info: Out of Memory: Not enough space to allocate {} B {} buffer across {} banks, where each bank needs to store {} B 44841600 L1 75 597888
b) TT_THROW @ tt_metal/impl/program/program.cpp:491: tt::exception info: Statically allocated circular buffers in program {} clash with L1 buffers on core range {}. L1 buffer allocated at {} and static circular buffer region ends at {} 978 [(x=0,y=0) - (x=11,y=7)] 236992 474464
tests/ttnn/unit_tests/operations/test_conv2d.py:175: AssertionError
tt_eager/tt_dnn/op_library/sliding_window_op_infra/tt_py_composite_conv.py:258: AssertionError
commit for the mobilenetv2 ops unit tests is here
PR: #6864
To create unittests for all the ops in the MobilenetV2 model on GS device. unit tests to be created under https://github.com/tenstorrent-metal/tt-metal/blob/main/tests/ttnn/unit_tests/operations