Skip to content

Add pooling and softmax unittests for Arm backend #2645

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

Closed

Conversation

oscarandersson8218
Copy link
Collaborator

  • Add SoftMax unittests
  • Add mean_dim unittests
  • Add AvgPool2d unittests
  • Enable linear u55 test

Copy link

pytorch-bot bot commented Mar 25, 2024

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/2645

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit fc4decd with merge base 853fb5b (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Mar 25, 2024
@oscarandersson8218 oscarandersson8218 added partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm ciflow/trunk and removed CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. labels Mar 25, 2024
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Mar 25, 2024
return self.softmax(x)

@register_test
# @register_test
Copy link
Contributor

Choose a reason for hiding this comment

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

Is this intentional?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

No, pushed a new patch reverting that change.

@facebook-github-bot
Copy link
Contributor

@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@digantdesai
Copy link
Contributor

Rebase please?

* Add SoftMax unittests
* Add mean_dim unittests
* Add AvgPool2d unittests
* Enable linear u55 test

Change-Id: I0a1f9f1b0f1fea13a2c9727b6b57f09fff8da5eb
Signed-off-by: Oscar Andersson <[email protected]>
Signed-off-by: Oscar Andersson <[email protected]>
Change-Id: I81b431f9de2b151d61bf93916370ff4d948bcd97
@facebook-github-bot
Copy link
Contributor

@digantdesai has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@facebook-github-bot
Copy link
Contributor

@digantdesai merged this pull request in 65f3e18.

kirklandsign pushed a commit to kirklandsign/executorch that referenced this pull request Apr 4, 2024
Summary:
* Add SoftMax unittests
* Add mean_dim unittests
* Add AvgPool2d unittests
* Enable linear u55 test

Pull Request resolved: pytorch#2645

Reviewed By: mergennachin

Differential Revision: D55316136

Pulled By: digantdesai

fbshipit-source-id: 80d9a2e34a786742b47cb08b9b9c640f6b61964d
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ciflow/trunk CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Merged partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants