Skip to content

Add portable var_mean.correction_out kernel (#18775)#18775

Merged
meta-codesync[bot] merged 1 commit intopytorch:mainfrom
pssrawat:export-D100016876
Apr 9, 2026
Merged

Add portable var_mean.correction_out kernel (#18775)#18775
meta-codesync[bot] merged 1 commit intopytorch:mainfrom
pssrawat:export-D100016876

Conversation

@pssrawat
Copy link
Copy Markdown
Contributor

@pssrawat pssrawat commented Apr 8, 2026

Summary:

Add a new portable ExecuTorch kernel for var_mean.correction_out that computes both variance and mean in a single function call. Uses the same two-pass algorithm as var.correction_out — the mean computed in pass 1 is written to the output tensor instead of being discarded.

Reviewed By: manuelcandales

Differential Revision: D100016876

@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 8, 2026
@meta-codesync
Copy link
Copy Markdown
Contributor

meta-codesync Bot commented Apr 8, 2026

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

@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented Apr 8, 2026

🔗 Helpful Links

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

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

✅ You can merge normally! (2 Unrelated Failures)

As of commit 5deccf8 with merge base 2d13fae (image):

FLAKY - The following job failed but was likely due to flakiness present on trunk:

BROKEN TRUNK - The following job failed but was 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.

@pssrawat pssrawat added the release notes: none Do not include this in the release notes label Apr 8, 2026
@meta-codesync meta-codesync Bot changed the title Add portable var_mean.correction_out kernel Add portable var_mean.correction_out kernel (#18775) Apr 8, 2026
pssrawat added a commit to pssrawat/executorch that referenced this pull request Apr 8, 2026
Summary:

Add a new portable ExecuTorch kernel for var_mean.correction_out that computes both variance and mean in a single function call. Uses the same two-pass algorithm as var.correction_out — the mean computed in pass 1 is written to the output tensor instead of being discarded.

Differential Revision: D100016876
@pssrawat pssrawat force-pushed the export-D100016876 branch from ff7b419 to e1ce260 Compare April 8, 2026 17:16
pssrawat added a commit to pssrawat/executorch that referenced this pull request Apr 8, 2026
Summary:

Add a new portable ExecuTorch kernel for var_mean.correction_out that computes both variance and mean in a single function call. Uses the same two-pass algorithm as var.correction_out — the mean computed in pass 1 is written to the output tensor instead of being discarded.

Reviewed By: manuelcandales

Differential Revision: D100016876
@pssrawat pssrawat force-pushed the export-D100016876 branch from e1ce260 to 9102af7 Compare April 8, 2026 19:35
pssrawat added a commit to pssrawat/executorch that referenced this pull request Apr 8, 2026
Summary:

Add a new portable ExecuTorch kernel for var_mean.correction_out that computes both variance and mean in a single function call. Uses the same two-pass algorithm as var.correction_out — the mean computed in pass 1 is written to the output tensor instead of being discarded.

Reviewed By: manuelcandales

Differential Revision: D100016876
@pssrawat pssrawat force-pushed the export-D100016876 branch from 9102af7 to f2f676c Compare April 8, 2026 19:36
pssrawat added a commit to pssrawat/executorch that referenced this pull request Apr 8, 2026
Summary:
Pull Request resolved: pytorch#18775

Add a new portable ExecuTorch kernel for var_mean.correction_out that computes both variance and mean in a single function call. Uses the same two-pass algorithm as var.correction_out — the mean computed in pass 1 is written to the output tensor instead of being discarded.

Reviewed By: manuelcandales

Differential Revision: D100016876
@pssrawat pssrawat force-pushed the export-D100016876 branch from f2f676c to 24e70e0 Compare April 8, 2026 19:39
Summary:

Add a new portable ExecuTorch kernel for var_mean.correction_out that computes both variance and mean in a single function call. Uses the same two-pass algorithm as var.correction_out — the mean computed in pass 1 is written to the output tensor instead of being discarded.

Reviewed By: manuelcandales

Differential Revision: D100016876
@pssrawat pssrawat force-pushed the export-D100016876 branch from 24e70e0 to 5deccf8 Compare April 8, 2026 20:28
@meta-codesync meta-codesync Bot merged commit a231fbc into pytorch:main Apr 9, 2026
161 of 165 checks passed
jpiat pushed a commit to jpiat/executorch that referenced this pull request Apr 14, 2026
Differential Revision: D100016876

Pull Request resolved: pytorch#18775
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. fb-exported meta-exported 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