Skip to content
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

Merged
merged 1 commit into from
Dec 9, 2024
Merged

Formating conv2d output #15741

merged 1 commit into from
Dec 9, 2024

Conversation

shwetankTT
Copy link
Contributor

@shwetankTT shwetankTT commented Dec 5, 2024

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

  • Post commit CI passes
  • Blackhole Post commit (if applicable)
  • Model regression CI testing passes (if applicable)
  • Device performance regression CI testing passes (if applicable)
  • New/Existing tests provide coverage for changes

Copy link
Contributor

@esmalTT esmalTT left a 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?

@shwetankTT
Copy link
Contributor Author

@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.
https://github.com/tenstorrent/tt-metal/actions/runs/12203020768/job/34046783895

@shwetankTT shwetankTT force-pushed the shwetankTT/conv_op_fix branch from 9b2af82 to 378f29d Compare December 6, 2024 19:26
@shwetankTT shwetankTT force-pushed the shwetankTT/conv_op_fix branch from 378f29d to ebc381a Compare December 7, 2024 01:40
Copy link
Member

@ayerofieiev-tt ayerofieiev-tt left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

No review, unblocking.

@shwetankTT shwetankTT force-pushed the shwetankTT/conv_op_fix branch 2 times, most recently from 4651c2a to 55fb873 Compare December 9, 2024 11:12
@shwetankTT shwetankTT force-pushed the shwetankTT/conv_op_fix branch from 55fb873 to faf91b3 Compare December 9, 2024 15:47
Copy link
Contributor

@dvartaniansTT dvartaniansTT left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@shwetankTT shwetankTT merged commit c12bf8e into main Dec 9, 2024
174 of 177 checks passed
@shwetankTT shwetankTT deleted the shwetankTT/conv_op_fix branch December 9, 2024 19:11
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

10 participants