Skip to content

Add optimized Arm64 BF16 _to_copy - #22493

Open
JakeStevens wants to merge 1 commit into
pytorch:mainfrom
JakeStevens:export-D118502073
Open

Add optimized Arm64 BF16 _to_copy#22493
JakeStevens wants to merge 1 commit into
pytorch:mainfrom
JakeStevens:export-D118502073

Conversation

@JakeStevens

Copy link
Copy Markdown
Contributor

Summary: Add an optimized _to_copy.out kernel. On Arm64, contiguous FP32-to-BF16 and BF16-to-FP32 conversions process eight elements per NEON iteration and use the ExecuTorch threadpool above its grain size. Other dtype pairs, layouts, and platforms fall back to the portable implementation, which remains unchanged.

Differential Revision: D118502073

Summary: Add an optimized `_to_copy.out` kernel. On Arm64, contiguous FP32-to-BF16 and BF16-to-FP32 conversions process eight elements per NEON iteration and use the ExecuTorch threadpool above its grain size. Other dtype pairs, layouts, and platforms fall back to the portable implementation, which remains unchanged.

Differential Revision: D118502073
@pytorch-bot

pytorch-bot Bot commented Sep 2, 2026

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 5 New Failures

As of commit 4ee4411 with merge base 4747ab7 (image):

NEW FAILURES - The following jobs have failed:

  • Cadence Build & Test / hifi-build / hifi4 (gh)
    ##[error]Refusing to check out fork pull request code from a 'pull_request_target' workflow. This workflow runs with the base repository's GITHUB_TOKEN, secrets, default-branch cache scope, and runner access. Fetching and executing a fork's code in that trusted context commonly leads to "pwn request" vulnerabilities. To opt in, review the risks at https://gh.io/securely-using-pull_request_target and set 'allow-unsafe-pr-checkout: true' on the actions/checkout step.
  • Cadence Build & Test / vision-build / vision (gh)
    ##[error]Refusing to check out fork pull request code from a 'pull_request_target' workflow. This workflow runs with the base repository's GITHUB_TOKEN, secrets, default-branch cache scope, and runner access. Fetching and executing a fork's code in that trusted context commonly leads to "pwn request" vulnerabilities. To opt in, review the risks at https://gh.io/securely-using-pull_request_target and set 'allow-unsafe-pr-checkout: true' on the actions/checkout step.
  • pull / unittest / linux / linux-job (gh)
    pytorch_tokenizers
  • pull / unittest-buck / linux / linux-job (gh)
    pytorch_tokenizers
  • pull / unittest-editable / linux / linux-job (gh)
    pytorch_tokenizers

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 Sep 2, 2026
@meta-codesync

meta-codesync Bot commented Sep 2, 2026

Copy link
Copy Markdown
Contributor

@JakeStevens has exported this pull request. If you are a Meta employee, you can view the originating Diff in D118502073.

@github-actions

github-actions Bot commented Sep 2, 2026

Copy link
Copy Markdown

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant