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

[TT-Train ]added tests for sum and mean #16152

Merged
merged 12 commits into from
Dec 19, 2024
Merged

Conversation

dmakoviichuk-tt
Copy link
Contributor

@dmakoviichuk-tt dmakoviichuk-tt commented Dec 18, 2024

Problem description

We need to know that ttnn reduce ops are almost the same as moreh.

What's changed

  • Added sum test
  • Added mean test
  • Updated sum_over_dim
Screenshot 2024-12-18 at 1 56 19 PM

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)
  • (For models and ops writers) Full new models tests passes
  • New/Existing tests provide coverage for changes

https://github.com/tenstorrent/tt-metal/actions/runs/12405523060

Copy link
Contributor

@rfurko-tt rfurko-tt left a comment

Choose a reason for hiding this comment

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

LGTM

@dmakoviichuk-tt dmakoviichuk-tt merged commit 5c91e97 into main Dec 19, 2024
182 of 184 checks passed
@dmakoviichuk-tt dmakoviichuk-tt deleted the DM/reduce_tt_train branch December 19, 2024 17:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants