Tests for transpose op - Githubissues

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

Tests for transpose op #711

Open kmilanovicTT opened 1 week ago

kmilanovicTT commented 1 week ago

Test torch transpose operator

Supported operand source models for Forge:

from dram from host from another op const eval pass mf and df tests

Closes #253

dgolubovicTT commented 3 days ago

Where can I see exact input shapes we use for testing transpose?

kmilanovicTT commented 3 days ago

Where can I see exact input shapes we use for testing transpose?

All input shapes used for testing are in file: tt-forge-fe/forge/test/operators/utils/test_data.py TestCollectionCommon.all.input_shapes.
Those input_shapes that are failing are described in 'failing_rules' in file tt-forge-fe/forge/test/operators/pytorch/tm/test_transpose.py:269