search

tenstorrent / tt-forge-fe

The TT-Forge FE is a graph compiler designed to optimize and transform computational graphs for deep learning models, enhancing their performance and efficiency.
https://docs.tenstorrent.com/tt-forge-fe/
Apache License 2.0
20 stars 3 forks source link

[Llama 3.2 1B] Support for reshape op (unhandled attribute type) #708

Closed mstojkovicTT closed 1 week ago

mstojkovicTT commented 1 week ago

When reshaping to 3D tensor, instead of casting shape to vector, shape gets cast to tuple<int,int,int> which is not handled.

Example: shapes: ((1, 11, 2048), (1, 11, 2048))

Repro: pytest forge/test/mlir/llama/tests/test_specific_ops_llama32.py::test_reshape

mstojkovicTT commented 1 week ago

This issue is resolved in PR #681