Skip to content

Arm backend: Add TOSA dialect ARGMAX node visitor#20418

Open
SaoirseARM wants to merge 3 commits into
pytorch:mainfrom
SaoirseARM:toupstream/arg_node
Open

Arm backend: Add TOSA dialect ARGMAX node visitor#20418
SaoirseARM wants to merge 3 commits into
pytorch:mainfrom
SaoirseARM:toupstream/arg_node

Conversation

@SaoirseARM

@SaoirseARM SaoirseARM commented Jun 22, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Adds ARGMAX to ExirToTosaPass
  • Uses _DIALECT_SUBSTITUTIONS for TOSA Activation ops

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell @rascani

- Use _DIALECT_SUBSTITUTIONS for activations

Change-Id: I7a4bb7a53789b0b86618686927a93840464b80ba
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
@SaoirseARM SaoirseARM requested a review from digantdesai as a code owner June 22, 2026 10:35
@pytorch-bot

pytorch-bot Bot commented Jun 22, 2026

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

❌ 2 New Failures

As of commit d9f70b0 with merge base 484f72c (image):

NEW FAILURES - The following jobs have failed:

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

@meta-cla meta-cla 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 Jun 22, 2026
@github-actions github-actions Bot added ciflow/trunk module: arm Issues related to arm backend labels Jun 22, 2026
@SaoirseARM

Copy link
Copy Markdown
Collaborator Author

@pytorchbot label ciflow/trunk

@SaoirseARM

Copy link
Copy Markdown
Collaborator Author

@pytorchbot label 'partner: arm'

@pytorch-bot pytorch-bot Bot added the partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm label Jun 22, 2026
@SaoirseARM

Copy link
Copy Markdown
Collaborator Author

@pytorchbot label "release notes: arm"

@pytorch-bot pytorch-bot Bot added the release notes: arm Changes to the ARM backend delegate label Jun 22, 2026

@oscarandersson8218 oscarandersson8218 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

OK to merge, just double check if any buck2 changes are needed due to the added files

Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
Change-Id: I7a4bb7a53789b0b86618686927a93840464b80ba
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. module: arm Issues related to arm backend partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm release notes: arm Changes to the ARM backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants