[digantdesai]

Stack from ghstack (oldest at bottom):

• [executorch][native-vulkan] Lower rms_norm without a weight on the Vulkan engine #25097
• -> [executorch][vulkan] Make the et_vk.rms_norm weight optional #25096
• [executorch][native-vulkan] Run bf16 methods in f32 on the Vulkan engine #25095
• [executorch][native] Record rms_norm op samples #25094

aten.rms_norm takes an optional weight, but et_vk.rms_norm required one, so ET-VK had no kernel for a norm without an elementwise affine. The schema now takes Tensor? weight, and its eager reference skips the multiply for a None weight. When the weight is None, add_rms_norm_node skips the weight prepack, binds only the input, and dispatches the new rms_norm_no_weight texture and buffer variants. The HAS_WEIGHT shader parameter defaults to false, which compiles out the weight binding and multiply, and the weighted variants set it, so they keep their names and their math.

The WebGPU rms_norm kernel always binds a weight, so the WebGPU partitioner leaves an et_vk.rms_norm with a None weight on CPU.

Authored with Claude Code.

Differential Revision: D131119349

NOTE FOR REVIEWERS: This PR has internal Meta-specific changes or comments, please review them on Phabricator!

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 2 New Failures

As of commit 3568c3c with merge base 925c1bd ():

NEW FAILURES - The following jobs have failed:

• pull / unittest / linux / linux-job (gh)
  RuntimeError: Command bash /__w/_temp/exec_script failed with exit code 1
• pull / unittest / macos / macos-job (gh)
  RuntimeError: Command bash /Users/ec2-user/runner/_work/_temp/exec_script failed with exit code 1

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

[github-actions]

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.