[ExecuTorch][WebGPU] Shared pipeline + dispatch-grid + vec4-align helpers (op-suite foundation)#20842
[ExecuTorch][WebGPU] Shared pipeline + dispatch-grid + vec4-align helpers (op-suite foundation)#20842JCNTH wants to merge 1 commit into
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/20842
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (1 Unrelated Failure)As of commit e8b7c8d with merge base aceeb40 ( BROKEN TRUNK - The following job 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. |
This PR needs a
|
Stack from ghstack (oldest at bottom):
Foundation for the vision/voice/seg WebGPU op suite: the shared helpers every op builds on. Adds to
WebGPUUtils.hthemake_compute_pipelinebundle (BindingSpec,ComputePipelineBundle,make_storage_zeros),check_vec4_aligned, and the validation/dispatch-config helpers every handler reuses:check_fp32andcheck_elementwise_fp32_io(fp32 byte-size + elementwise input/output guards),checked_u32(theuint64_t->uint32_tdispatch-count guard),make_wg_size_constant(the 1-element override-constant sibling ofmake_grid_constants), andcompute_tile_grid_2d(the tiled-GEMM 2D grid, device-limit-queried); a newWebGPUDispatchMath.hwith the near-square 2D dispatch-grid fold (div_up,numel); aWebGPUGraph::add_dispatch_2dconvenience for the 2D dispatch fill; sharedWebGPUGraphhooks; and op-test-driver infra (+test_webgpu_utils). No op is registered here — each op lands on top using these helpers.Co-authored-with: Claude Code.
Differential Revision: D110836666