-
Notifications
You must be signed in to change notification settings - Fork 6
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Convert aten.reshape
to ttnn.reshape
#190
base: main
Are you sure you want to change the base?
Conversation
Test started here |
Fails during models test
|
I just ran |
@jdh8 can you check this? |
Keep in mind that right now compiler Tilizes tensors before most TT-NN operations. |
75aa73b
to
f1ae5ec
Compare
@pytest.mark.parametrize( | ||
"input_shape, new_shape", | ||
[((3 * 5, 7), (7 * 3, 5))], | ||
(((3 * 5, 7), (7 * 3, 5)), ((1, 512, 1, 1), (1, 512))), |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This case passes on its own. Maybe we need to check the next op or the synergy between reshape
and the next op?
f1ae5ec
to
f86743c
Compare
Ticket
aten
tottnn
tt-metal#12853aten.view
not converted? #173Problem description
Successfully convert
aten.reshape
tottnn.reshape
This PR paves the way for #148, too.
What's changed
aten.reshape
tottnn.reshape
aten.view
tottnn.reshape
torch.reshape
tottnn.reshape