Skip to content

Arm backend: Add MOD_SHAPE backend dialect op#18836

Merged
oscarandersson8218 merged 2 commits intopytorch:mainfrom
oscarandersson8218:mod_shape
Apr 14, 2026
Merged

Arm backend: Add MOD_SHAPE backend dialect op#18836
oscarandersson8218 merged 2 commits intopytorch:mainfrom
oscarandersson8218:mod_shape

Conversation

@oscarandersson8218
Copy link
Copy Markdown
Collaborator

@oscarandersson8218 oscarandersson8218 commented Apr 13, 2026

Adds new TOSA backend dialect op for MOD_SHAPE. MOD_SHAPE computes arg0 modulo arg1, i.e:
def MOD_SHAPE(arg0, arg1):
out_shape = []
for dim0, dim1 in zip(arg0, arg1):
out_shape.append(dim0 % dim1)
return out_shape

Change-Id: I2f934eb251263a1973a0fc6c3d7723c8fb2a7bc1

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

Adds new TOSA backend dialect op for MOD_SHAPE. MOD_SHAPE computes arg0
modulo arg1, i.e:
def MOD_SHAPE(arg0, arg1):
    out_shape = []
    for dim0, dim1 in zip(arg0, arg1):
        out_shape.append(dim0 % dim1)
    return out_shape

Signed-off-by: Oscar Andersson <oscar.andersson@arm.com>
Change-Id: I2f934eb251263a1973a0fc6c3d7723c8fb2a7bc1
@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented Apr 13, 2026

🔗 Helpful Links

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

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:

❌ 6 New Failures, 3 Unrelated Failures

As of commit 8ad8815 with merge base 37b12c8 (image):

NEW FAILURES - The following jobs have failed:

BROKEN TRUNK - The following jobs failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

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 Apr 13, 2026
@oscarandersson8218 oscarandersson8218 added partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm ciflow/trunk release notes: none Do not include this in the release notes labels Apr 13, 2026
@oscarandersson8218
Copy link
Copy Markdown
Collaborator Author

Unrelated failures.

@oscarandersson8218 oscarandersson8218 merged commit 2c545f8 into pytorch:main Apr 14, 2026
410 of 422 checks passed
jpiat pushed a commit to jpiat/executorch that referenced this pull request Apr 14, 2026
Adds new TOSA backend dialect op for MOD_SHAPE. MOD_SHAPE computes arg0
modulo arg1, i.e:
def MOD_SHAPE(arg0, arg1):
    out_shape = []
    for dim0, dim1 in zip(arg0, arg1):
        out_shape.append(dim0 % dim1)
    return out_shape


Change-Id: I2f934eb251263a1973a0fc6c3d7723c8fb2a7bc1


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

Signed-off-by: Oscar Andersson <oscar.andersson@arm.com>
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. partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm release notes: none Do not include this in the release notes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants