-
Notifications
You must be signed in to change notification settings - Fork 96
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
Formating conv2d output #15741
Formating conv2d output #15741
Conversation
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.
I am seeing failures in Mamba on the nightly pipeline that don't appear on main: https://github.com/tenstorrent/tt-metal/actions/runs/12180236565/job/33990661239#step:9:964
Can you rebase with latest and try again?
0af184d
to
009aecf
Compare
@esmalTT Hey Evan Can you please unblock me so that I can merge the PR over the weekend. I am running all the pipelines again. Hopefully they pass. |
9b2af82
to
378f29d
Compare
378f29d
to
ebc381a
Compare
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.
No review, unblocking.
4651c2a
to
55fb873
Compare
Co-authored-by: Lewis Panos <[email protected]>
55fb873
to
faf91b3
Compare
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.
LGTM
Ticket
#14179
Problem description
Parameterized conv2d output from the Python side:
Users can select the desired output. By default, the function returns the convolution output (conv_output). However, users have the option to receive additional outputs, including the output dimensions, weights, and biases, depending on their needs.
cc: @LPanosTT
(Single-card) Demo tests
(Single-card) Device perf regressions
(Single-card) Model perf tests
Nightly fast dispatch tests
Checklist