[ExecuTorch][WebGPU] Add select/boolean ops (where, scalar compares, logical_not) to the WebGPU backend#20925
[ExecuTorch][WebGPU] Add select/boolean ops (where, scalar compares, logical_not) to the WebGPU backend#20925JCNTH wants to merge 3 commits into
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/20925
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 3ec140c with merge base 7013c8d ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
SS-JIA
left a comment
There was a problem hiding this comment.
Review automatically exported from Phabricator review in Meta.
Stack from ghstack (oldest at bottom):
Add
aten.where.self, scalar-comparison (aten.{eq,ne,le,ge,lt}.Scalar), andaten.logical_not.defaulthandlers for the training tail.Key changes:
runtime/ops/{where,compare,logical_not}/— WGSL kernels + handlersop_registry.py— register the net-newaten.{ne,lt,le,ge}.Scalar+aten.logical_not.defaultOpFeatures(eq.Scalaralready on master)CMakeLists.txtWEBGPU_SRCS— wire the sourcesaten.where.selfalready had VulkanOpFeatures; the scalar-compare +logical_notregistrations are additive to the shared Vulkan registry (byte-packed-bool WGSL kernels).Co-authored-with: Claude Code.
@exported-using-ghexport
Differential Revision: D111755122
Differential Revision: D111755122