[ExecuTorch][WebGPU] Migrate 9 ops to make_compute_pipeline (layout:"auto")#20868
[ExecuTorch][WebGPU] Migrate 9 ops to make_compute_pipeline (layout:"auto")#20868JCNTH wants to merge 1 commit into
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/20868
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 New Failure, 2 Unrelated FailuresAs of commit 5e7ab80 with merge base aceeb40 ( NEW FAILURE - The following job has failed:
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. |
This PR needs a
|
Stack from ghstack (oldest at bottom):
Route
add,mul,index,permute,select,slice,update_cache,rms_norm, andembedding_q4gswthrough the sharedutils::make_compute_pipelinehelper (which useslayout:"auto"), replacing each op's hand-writtenWGPUBindGroupLayoutEntry[]+ pipeline-layout + bind-group boilerplate (~50-110 lines each) with a single helper call. The driver now derives the bind-group layout from the shader's statically-used bindings. Byte-behavior is preserved: identical binding indices/types/buffers/sizes, dispatch workgroup counts, resize hooks, and validations; override constants (wg_size) passed via the helper'sconstantsparam. Extends the Diff 1 layout:"auto" adoption to the trivial single-dispatch ops.Differential Revision: D110836665